wanghanrui commited on
Commit
8fd1afe
·
1 Parent(s): 765bf08

Comment out the _supports_sdpa property in TinyLlavaPreTrainedModel

Browse files
Files changed (1) hide show
  1. modeling_mufun.py +3 -3
modeling_mufun.py CHANGED
@@ -226,9 +226,9 @@ class TinyLlavaPreTrainedModel(PreTrainedModel):
226
  if module.padding_idx is not None:
227
  module.weight.data[module.padding_idx].zero_()
228
 
229
- @property
230
- def _supports_sdpa(self):
231
- return self.language_model._supports_sdpa
232
 
233
 
234
  class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel, GenerationMixin):
 
226
  if module.padding_idx is not None:
227
  module.weight.data[module.padding_idx].zero_()
228
 
229
+ # @property
230
+ # def _supports_sdpa(self):
231
+ # return self.language_model._supports_sdpa
232
 
233
 
234
  class TinyLlavaForConditionalGeneration(TinyLlavaPreTrainedModel, GenerationMixin):