forgot to remove debug messages
This commit is contained in:
parent
9c9f048b5e
commit
036a0dde70
4
webui.py
4
webui.py
|
@ -211,13 +211,9 @@ def setup_for_low_vram(sd_model):
|
||||||
return
|
return
|
||||||
|
|
||||||
if module_in_gpu is not None:
|
if module_in_gpu is not None:
|
||||||
print('removing from gpu:', type(module_in_gpu))
|
|
||||||
module_in_gpu.to(cpu)
|
module_in_gpu.to(cpu)
|
||||||
|
|
||||||
print('adding to gpu:', type(module))
|
|
||||||
module.to(gpu)
|
module.to(gpu)
|
||||||
|
|
||||||
print('added to gpu:', type(module))
|
|
||||||
module_in_gpu = module
|
module_in_gpu = module
|
||||||
|
|
||||||
# see below for register_forward_pre_hook;
|
# see below for register_forward_pre_hook;
|
||||||
|
|
Loading…
Reference in New Issue
Block a user