ceos/unsloth
1
0
Fork 0
Code Issues Pull requests Projects Releases Packages Wiki Activity Actions

Update vision.py

Browse source
This commit is contained in:
Daniel Han 2025-03-18 22:33:18 -07:00
commit 369ce004df
1 changed files with 4 additions and 10 deletions
Download patch file Download diff file Expand all files Collapse all files

14
unsloth/models/vision.py
View file

@ -81,7 +81,6 @@ def unsloth_base_fast_generate(
*args,
**kwargs,
):
print(args, kwargs)
if len(args) != 0:
x = args[0]
elif "input_ids" in kwargs:
@ -544,15 +543,10 @@ class FastBaseModel:
model.for_inference = functools.partial(FastBaseModel.for_inference, model)
# Patch generate
# if model.generate.__name__ != "unsloth_base_fast_generate":
# # Check for internal old_generates
# m = model
# while hasattr(m, "model"):
# if hasattr(m, "_old_generate"):
# model._old_generate = model.generate
# unsloth_base_fast_generate.__doc__ = model._old_generate.__doc__
# model.generate = types.MethodType(unsloth_base_fast_generate, model)
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)
return model
pass