Remove print.

708138c7 · comfyanonymous · 04777561 · 708138c7
Commit 708138c7 authored Feb 08, 2023 by comfyanonymous
Hide whitespace changes
Inline Side-by-side

Showing with 1 addition and 1 deletion

comfy/model_management.py comfy/model_management.py +1 -1

No files found.
--- a/comfy/model_management.py
+++ b/comfy/model_management.py
@@ -84,7 +84,7 @@ def load_model_gpu(model):
            device_map = accelerate.infer_auto_device_map(real_model, max_memory={0: "256MiB", "cpu": "16GiB"})
        elif vram_state == LOW_VRAM:
            device_map = accelerate.infer_auto_device_map(real_model, max_memory={0: "{}MiB".format(total_vram_available_mb), "cpu": "16GiB"})
-        print(device_map, "{}MiB".format(total_vram_available_mb))
+
        accelerate.dispatch_model(real_model, device_map=device_map, main_device="cuda")
        model_accelerated = True
    return current_loaded_model