Skip to content

GitLab

"...git@developer.sourcefind.cn:chenpangpang/ComfyUI.git" did not exist on "bed116a1f916f23b65c7b68cf60d09b2401fcf29"
Commit 708138c7 authored by comfyanonymous's avatar comfyanonymous
Browse files

Remove print.

parent 04777561
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
comfy/model_management.py
View file @ 708138c7
......@@ -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
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment