SafaaAI commited on
Commit
6be3aab
·
verified ·
1 Parent(s): 7380e1e

Update modeling_tinyllava_phi.py

Browse files
Files changed (1) hide show
  1. modeling_tinyllava_phi.py +4 -4
modeling_tinyllava_phi.py CHANGED
@@ -284,10 +284,10 @@ class TinyLlavaPreTrainedModel(PreTrainedModel):
284
 
285
  class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel):
286
  def __init__(self, config: TinyLlavaConfig):
287
- # Cette ligne est la correction. Nous créons un dictionnaire
288
- # vide et le passons comme argument. La valeur de `_supports_sdpa`
289
- # est lue depuis ce dictionnaire.
290
- config.vision_config._supports_sdpa = True
291
  super().__init__(config)
292
 
293
  self.language_model = PhiForCausalLM(config.text_config)
 
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)