Update fast_lora.py
This commit is contained in:
parent
a0a409d5af
commit
c4f0de58a9
1 changed files with 1 additions and 1 deletions
|
|
@ -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
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue