use __func__ to check can_generate (#22643)

use func to check can_generate (#22643)
f74b4020 · xinhe · GitHub · 14fc1a24 · f74b4020
Unverified Commit f74b4020 authored Apr 10, 2023 by xinhe Committed by GitHub Apr 10, 2023
Show whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

src/transformers/modeling_utils.py src/transformers/modeling_utils.py +1 -1

No files found.
--- a/src/transformers/modeling_utils.py
+++ b/src/transformers/modeling_utils.py
@@ -1185,7 +1185,7 @@ class PreTrainedModel(nn.Module, ModuleUtilsMixin, GenerationMixin, PushToHubMix
            `bool`: Whether this model can generate sequences with `.generate()`.
        """
        # Detects whether `prepare_inputs_for_generation` has been overwritten, which is a requirement for generation
-        if "GenerationMixin" in str(self.prepare_inputs_for_generation):
+        if "GenerationMixin" in str(self.prepare_inputs_for_generation.__func__):
            return False
        return True