From 544195168ca2fad8175dbfa3cb3659a22ce23f50 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Thu, 16 Apr 2026 15:35:09 +0000 Subject: [PATCH] style: shorten call-site comment in vision.py --- unsloth/models/vision.py | 6 +----- 1 file changed, 1 insertion(+), 5 deletions(-) diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 85aa4b62b9..ba23a575c8 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -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,