Upload model

config.json

@@ -6,15 +6,29 @@
     "AutoConfig": "configuration_musilingo.MusiLingoConfig",
     "AutoModel": "modelling_musilingo.MusilingoModel"
   },
+  "bos_token_id": 1,
   "device_8bit": 0,
   "end_sym": "\n",
-  "llama_model": "/data/home/acw688/MusiLingo/model/7B_vicuna",
+  "eos_token_id": 2,
+  "hidden_act": "silu",
+  "hidden_size": 4096,
+  "initializer_range": 0.02,
+  "intermediate_size": 11008,
+  "llama_model": "lmsys/vicuna-7b-delta-v0",
   "low_resource": false,
+  "max_position_embeddings": 2048,
   "max_txt_len": 32,
   "mert_model": "m-a-p/MERT-v1-330M",
   "model_type": "musilingo",
+  "num_attention_heads": 32,
+  "num_hidden_layers": 32,
+  "pad_token_id": 0,
   "prompt_path": "",
   "prompt_template": "###Human: {} ###Assistant: ",
+  "rms_norm_eps": 1e-06,
+  "tie_word_embeddings": false,
   "torch_dtype": "float32",
-  "transformers_version": "4.39.3"
+  "transformers_version": "4.39.3",
+  "use_cache": true,
+  "vocab_size": 32001
 }

modelling_musilingo.py

@@ -2019,6 +2019,7 @@ class MusiLingo(BaseModel):
         self,
         mert_model,
         llama_model,
+        config,
         prompt_path="",
         prompt_template="",
         max_txt_len=32,
@@ -2259,6 +2260,7 @@ class MusilingoModel(PreTrainedModel):
         self.model = MusiLingo(
             mert_model=config.mert_model,
             llama_model=config.llama_model,
+            config=config,
             prompt_path=config.prompt_path,
             prompt_template=config.prompt_template,
             max_txt_len=config.max_txt_len,