Commit 036a0dde authored by AUTOMATIC's avatar AUTOMATIC

forgot to remove debug messages

parent 9c9f048b
......@@ -211,13 +211,9 @@ def setup_for_low_vram(sd_model):
return
if module_in_gpu is not None:
print('removing from gpu:', type(module_in_gpu))
module_in_gpu.to(cpu)
print('adding to gpu:', type(module))
module.to(gpu)
print('added to gpu:', type(module))
module_in_gpu = module
# see below for register_forward_pre_hook;
......
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