Arthur-LAGACHERIE commited on
Commit
94e77ae
·
verified ·
1 Parent(s): 1c10bfd

add mutiple layers in MetaLoRALayer

Browse files
Files changed (1) hide show
  1. configuration_metalora.py +2 -0
configuration_metalora.py CHANGED
@@ -10,6 +10,7 @@ class MetaLoRAConfig(PretrainedConfig):
10
  base_size:int=384,
11
  embd_model:str="init",
12
  llm_tokenizer:str="init",
 
13
  **kwargs,
14
  ):
15
  self.auto_map = {"AutoModel": "Arthur-LAGACHERIE/MetaLoRA-code--modeling_metalora.MLoRAModel", "AutoModelForCausalLM": "Arthur-LAGACHERIE/MetaLoRA-code--modeling_metalora.MLoRAModel", "AutoConfig":"Arthur-LAGACHERIE/MetaLoRA-code--configuration_metalora.MetaLoRAConfig"}
@@ -17,4 +18,5 @@ class MetaLoRAConfig(PretrainedConfig):
17
  self.base_size = base_size
18
  self.embd_model = embd_model
19
  self.llm_tokenizer = llm_tokenizer
 
20
  super().__init__(**kwargs)
 
10
  base_size:int=384,
11
  embd_model:str="init",
12
  llm_tokenizer:str="init",
13
+ hidden_size_coef:float=0.5,
14
  **kwargs,
15
  ):
16
  self.auto_map = {"AutoModel": "Arthur-LAGACHERIE/MetaLoRA-code--modeling_metalora.MLoRAModel", "AutoModelForCausalLM": "Arthur-LAGACHERIE/MetaLoRA-code--modeling_metalora.MLoRAModel", "AutoConfig":"Arthur-LAGACHERIE/MetaLoRA-code--configuration_metalora.MetaLoRAConfig"}
 
18
  self.base_size = base_size
19
  self.embd_model = embd_model
20
  self.llm_tokenizer = llm_tokenizer
21
+ self.hidden_size_coef = hidden_size_coef
22
  super().__init__(**kwargs)