We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 5a5cfc7 commit 49b053cCopy full SHA for 49b053c
unsloth/models/vision.py
@@ -435,7 +435,6 @@ def from_pretrained(
435
if do_forced_float32: torch_dtype = torch.bfloat16
436
437
raise_handler = RaiseUninitialized()
438
- print(kwargs, model_name)
439
model = auto_model.from_pretrained(
440
model_name,
441
device_map = device_map,
@@ -446,7 +445,6 @@ def from_pretrained(
446
445
# attn_implementation = attn_implementation,
447
**kwargs,
448
)
449
- print(model)
450
raise_handler.remove()
451
# Return old flag
452
os.environ["HF_HUB_ENABLE_HF_TRANSFER"] = old_hf_transfer
0 commit comments