فهرست منبع

fix: transformers 4.54.0 adaptation

myhloli 3 ماه پیش
والد
کامیت
4aef8fd168
1فایلهای تغییر یافته به همراه5 افزوده شده و 4 حذف شده
  1. 5 4
      mineru/model/mfr/unimernet/unimernet_hf/unimer_mbart/modeling_unimer_mbart.py

+ 5 - 4
mineru/model/mfr/unimernet/unimernet_hf/unimer_mbart/modeling_unimer_mbart.py

@@ -1506,10 +1506,11 @@ class UnimerMBartDecoder(UnimerMBartPreTrainedModel):
                     continue
 
             # past_key_value = past_key_values[idx] if past_key_values is not None else None
-            past_key_value = None
-            if past_key_values is not None and len(past_key_values) > idx:
-                if isinstance(past_key_values, (list, tuple)):
-                    past_key_value = past_key_values[idx]
+            past_key_value = past_key_values[idx] if (
+                    past_key_values is not None and
+                    isinstance(past_key_values, (list, tuple)) and
+                    idx < len(past_key_values)
+            ) else None
 
             if self.gradient_checkpointing and self.training:
                 layer_outputs = self._gradient_checkpointing_func(