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

[pre-commit.ci] auto fixes from pre-commit.com hooks

Browse source 
for more information, see https://pre-commit.ci
This commit is contained in:
pre-commit-ci[bot] 2026-02-11 05:05:07 +00:00
commit 40de1d6407
1 changed files with 5 additions and 3 deletions
Download patch file Download diff file Expand all files Collapse all files

8
unsloth/models/_utils.py
View file

@ -2734,11 +2734,13 @@ def make_fast_generate_wrapper(original_generate):
"Unsloth: Passing text strings to `fast_generate` on AMD "
"requires a tokenizer attached to the model."
)
texts = [first_arg] if isinstance(first_arg, str) else list(first_arg)
texts = (
[first_arg] if isinstance(first_arg, str) else list(first_arg)
)
tokens = tokenizer(
texts,
return_tensors="pt",
padding=True,
return_tensors = "pt",
padding = True,
)
tokens = tokens.to(model.device)
return original_generate(**tokens, **kwargs)