Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-19 17:24:32 +02:00
Code Issues Packages Projects Releases Wiki Activity

Send to CPU intead of XPU when unloading

Browse Source
This commit is contained in:
Disty0
2023-05-24 13:03:40 +03:00
parent 5c516332d7
commit 0412651a6c
1 changed files with 6 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+6 -1
View File
@@ -465,6 +465,8 @@ def reload_model_weights(sd_model=None, info=None):
return
if shared.cmd_opts.lowvram or shared.cmd_opts.medvram:
lowvram.send_everything_to_cpu()
elif shared.cmd_opts.use_ipex:
model_data.sd_model.to("cpu")
else:
sd_model.to(devices.cpu)
sd_hijack.model_hijack.undo_hijack(sd_model)
@@ -501,7 +503,10 @@ def reload_model_weights(sd_model=None, info=None):
def unload_model_weights(sd_model=None, _info=None):
from modules import sd_hijack
if model_data.sd_model:
model_data.sd_model.to(devices.cpu)
if shared.cmd_opts.use_ipex:
model_data.sd_model.to("cpu")
else:
model_data.sd_model.to(devices.cpu)
sd_hijack.model_hijack.undo_hijack(model_data.sd_model)
model_data.sd_model = None
sd_model = None