Fix for transformers update

This commit is contained in:
kijai
2025-08-01 18:31:12 +03:00
parent de485b65b3
commit 6b3064b8bd
2 changed files with 7 additions and 2 deletions
+6 -2
View File
@@ -2340,7 +2340,9 @@ class Florence2PreTrainedModel(PreTrainedModel):
Retrieve language_model's attribute to check whether the model supports
Flash Attention 2 or not.
"""
return self.language_model._supports_flash_attn_2
if hasattr(self, 'language_model') and self.language_model is not None:
return self.language_model._supports_flash_attn_2
return True # Default to True during initialization
@property
def _supports_sdpa(self):
@@ -2348,7 +2350,9 @@ class Florence2PreTrainedModel(PreTrainedModel):
Retrieve language_model's attribute to check whether the model supports
SDPA or not.
"""
return self.language_model._supports_sdpa
if hasattr(self, 'language_model') and self.language_model is not None:
return self.language_model._supports_sdpa
return True # Default to True during initialization
FLORENCE2_INPUTS_DOCSTRING = r"""
+1
View File
@@ -384,6 +384,7 @@ class Florence2Run:
max_new_tokens=max_new_tokens,
do_sample=do_sample,
num_beams=num_beams,
use_cache=False
)
results = processor.batch_decode(generated_ids, skip_special_tokens=False)[0]