models without can_generate

huggingface · Sep 26, 2024 · 24b5a46 · 24b5a46
1 parent 2ce3723
commit 24b5a46
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/src/transformers/modeling_utils.py b/src/transformers/modeling_utils.py
@@ -1647,6 +1647,8 @@ def can_generate(cls) -> bool:
             return True
         # The class inherits from a class that can generate (recursive check) -> can generate
         for base in cls.__bases__:
+            if not hasattr(base, "can_generate"):
+                continue
             if "PreTrainedModel" not in str(base) and base.can_generate():
                 return True
         # BC: Detects whether `prepare_inputs_for_generation` has been overwritten in the model. Prior to v4.45, this