forgot to remove debug messages

This commit is contained in:
AUTOMATIC 2022-08-29 02:17:41 +03:00
parent 9c9f048b5e
commit 036a0dde70

View File

@ -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;