From b58c264e05cd9581c419578a8a1ffbafcba980ad Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 25 Mar 2025 23:20:00 -0700 Subject: [PATCH] Update vision.py --- unsloth/models/vision.py | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index ad0aeb9915..ef32ab1847 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -431,12 +431,11 @@ class FastBaseModel: m.is_loaded_in_8bit = True if not full_finetuning else False # Patch generate - if os.environ.get("UNSLOTH_DISABLE_FAST_GENERATION", "0") == "0": - if model.generate.__name__ != "unsloth_base_fast_generate": - model._old_generate = model.generate - unsloth_base_fast_generate.__doc__ = model._old_generate.__doc__ - model.generate = types.MethodType(unsloth_base_fast_generate, model) - pass + if model.generate.__name__ != "unsloth_base_fast_generate": + model._old_generate = model.generate + unsloth_base_fast_generate.__doc__ = model._old_generate.__doc__ + model.generate = types.MethodType(unsloth_base_fast_generate, model) + # Post patches model = FastBaseModel.post_patch_model( model,