SafaaAI commited on
Commit
7211c14
·
verified ·
1 Parent(s): 6be3aab

Update modeling_tinyllava_phi.py

Browse files
Files changed (1) hide show
  1. modeling_tinyllava_phi.py +2 -4
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)