Update modeling_tinyllava_phi.py
Browse files
modeling_tinyllava_phi.py
CHANGED
|
@@ -286,6 +286,7 @@ class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel):
|
|
| 286 |
def __init__(self, config: TinyLlavaConfig):
|
| 287 |
|
| 288 |
super().__init__(config)
|
|
|
|
| 289 |
|
| 290 |
self.language_model = PhiForCausalLM(config.text_config)
|
| 291 |
self.vision_tower = VisionTower(config.vision_config, config.vision_model_name_or_path)
|
|
|
|
| 286 |
def __init__(self, config: TinyLlavaConfig):
|
| 287 |
|
| 288 |
super().__init__(config)
|
| 289 |
+
self._supports_sdpa = True # Ligne ajoutée pour la compatibilité
|
| 290 |
|
| 291 |
self.language_model = PhiForCausalLM(config.text_config)
|
| 292 |
self.vision_tower = VisionTower(config.vision_config, config.vision_model_name_or_path)
|