Update modeling_tinyllava_phi.py
Browse files
modeling_tinyllava_phi.py
CHANGED
|
@@ -283,11 +283,9 @@ class TinyLlavaPreTrainedModel(PreTrainedModel):
|
|
| 283 |
|
| 284 |
|
| 285 |
class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel):
|
|
|
|
|
|
|
| 286 |
def __init__(self, config: TinyLlavaConfig):
|
| 287 |
-
# Cette ligne définit l'attribut directement sur le dictionnaire de l'instance.
|
| 288 |
-
# Cela contourne la protection qui empêche la définition normale de l'attribut.
|
| 289 |
-
self.__dict__['_supports_sdpa'] = True
|
| 290 |
-
|
| 291 |
super().__init__(config)
|
| 292 |
|
| 293 |
self.language_model = PhiForCausalLM(config.text_config)
|
|
|
|
| 283 |
|
| 284 |
|
| 285 |
class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel):
|
| 286 |
+
_supports_sdpa = True # Définissez l'attribut ici, au niveau de la classe.
|
| 287 |
+
|
| 288 |
def __init__(self, config: TinyLlavaConfig):
|
|
|
|
|
|
|
|
|
|
|
|
|
| 289 |
super().__init__(config)
|
| 290 |
|
| 291 |
self.language_model = PhiForCausalLM(config.text_config)
|