Update configuration.py

configuration.py

@@ -1,6 +1,7 @@
 from transformers import PretrainedConfig
 from transformers import CONFIG_MAPPING
 from transformers import AutoConfig
+from transformers.models.phi.configuration_phi import PhiConfig # Ligne ajoutée
 
 IGNORE_INDEX = -100
 IMAGE_TOKEN_INDEX = -200
@@ -79,9 +80,9 @@ class TinyLlavaConfig(PretrainedConfig):
     def _load_text_config(self, text_config=None):
         if self.llm_model_name_or_path is None or self.llm_model_name_or_path == '':
             self.text_config = CONFIG_MAPPING['llama']()
-           
+            
         else:
-            self.text_config = AutoConfig.from_pretrained(self.llm_model_name_or_path, trust_remote_code=True)
+            self.text_config = PhiConfig.from_pretrained(self.llm_model_name_or_path, trust_remote_code=True) # Ligne modifiée
             if text_config is not None:
                 self.text_config = self.text_config.from_dict(text_config)
                 
@@ -111,6 +112,4 @@ class TinyLlavaConfig(PretrainedConfig):
                 
         self.vision_config.model_name_or_path = self.vision_model_name_or_path.split(':')[-1]
         self.vision_config.model_name_or_path2 = self.vision_model_name_or_path2.split(':')[-1]
-        self.vision_hidden_size = getattr(self.vision_config, 'hidden_size',  None)  
-        
-
+        self.vision_hidden_size = getattr(self.vision_config, 'hidden_size',  None)