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

style: shorten call-site comment in vision.py

Browse source
This commit is contained in:
Daniel Han 2026-04-16 15:35:09 +00:00
commit 544195168c
1 changed files with 1 additions and 5 deletions
Download patch file Download diff file Expand all files Collapse all files

6
unsloth/models/vision.py
View file

@ -961,11 +961,7 @@ class FastBaseModel:
# attn_implementation = attn_implementation,
**kwargs,
)
# Attach AlignDevicesHook for bnb multi-device / non-default-device
# loads. The bnb loading path places weights on CUDA devices but
# never calls dispatch_model, so no hooks are installed. Without
# hooks, any cross-device module call crashes at forward time with
# "Expected all tensors to be on the same device".
# Attach dispatch hooks for bnb multi-device loads.
_attach_bnb_multidevice_hooks(
model,
load_in_4bit = load_in_4bit,