Skip to content

Commit 49b053c

Browse files
committed
Update vision.py
1 parent 5a5cfc7 commit 49b053c

File tree

1 file changed

+0
-2
lines changed

1 file changed

+0
-2
lines changed

unsloth/models/vision.py

Lines changed: 0 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -435,7 +435,6 @@ def from_pretrained(
435435
if do_forced_float32: torch_dtype = torch.bfloat16
436436

437437
raise_handler = RaiseUninitialized()
438-
print(kwargs, model_name)
439438
model = auto_model.from_pretrained(
440439
model_name,
441440
device_map = device_map,
@@ -446,7 +445,6 @@ def from_pretrained(
446445
# attn_implementation = attn_implementation,
447446
**kwargs,
448447
)
449-
print(model)
450448
raise_handler.remove()
451449
# Return old flag
452450
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = old_hf_transfer

0 commit comments

Comments
 (0)