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-05-16 23:03:04 -07:00
commit f130fba1b2
1 changed files with 1 additions and 0 deletions
Download patch file Download diff file Expand all files Collapse all files

1
unsloth/models/vision.py
View file

@ -89,6 +89,7 @@ def unsloth_base_fast_generate(
*args,
**kwargs,
):
print(args, kwargs)
if len(args) != 0:
input_ids = args[0]
elif "input_ids" in kwargs: