Update fast_lora.py

This commit is contained in:
Daniel Han-Chen 2024-01-26 19:21:00 +11:00
commit c4f0de58a9

View file

@ -178,7 +178,7 @@ pass
from transformers.models.llama.modeling_llama import logger
def apply_lora_mlp(self, X):
logger.warning_once("Hello!2")
logger.warning_once("Hello!3")
# gate = self.gate_proj(X)
# up = self. up_proj(X)
# h = torch.nn.functional.silu(gate) * up