Upload FloSmolV

modeling_flosmolv.py

@@ -11,15 +11,15 @@ class FloSmolV(PreTrainedModel):
     def __init__(self, config: FloSmolVConfig):
         super().__init__(config)
         self.florence2_model = AutoModelForCausalLM.from_pretrained(
-            self.config.vision_config._name_or_path,
+            self.config.vision_config["_name_or_path"],
             trust_remote_code=True,
         ).eval().to(self.device)
-        self.florence2_processor = AutoProcessor.from_pretrained(self.config.vision_config._name_or_path, trust_remote_code=True,)
+        self.florence2_processor = AutoProcessor.from_pretrained(self.config.vision_config["_name_or_path"], trust_remote_code=True,)
         self.smollm_model = AutoModelForCausalLM.from_pretrained(
-            self.config.llm_config._name_or_path,
+            self.config.llm_config["_name_or_path"],
             trust_remote_code=True,
         ).to(self.device)
-        self.smollm_tokenizer = AutoTokenizer.from_pretrained(self.config.llm_config._name_or_path, trust_remote_code=True,)
+        self.smollm_tokenizer = AutoTokenizer.from_pretrained(self.config.llm_config["_name_or_path"], trust_remote_code=True,)
       
     def forward(self, image, query: str):
       ## Vision