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

Revert xpu.optimize in training

Browse Source
This commit is contained in:
Disty0
2023-06-13 15:36:41 +03:00
parent f510abed39
commit 0d101b9def
2 changed files with 2 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/hypernetworks/hypernetwork.py
+1 -2
View File
@@ -590,8 +590,7 @@ def train_hypernetwork(id_task, hypernetwork_name, learn_rate, batch_size, gradi
print(e)
if shared.cmd_opts.use_ipex:
shared.sd_model.train()
shared.sd_model, optimizer = torch.xpu.optimize(shared.sd_model.to(dtype=torch.float32), optimizer=optimizer, dtype=devices.dtype)
pass
else:
scaler = torch.cuda.amp.GradScaler()
modules/textual_inversion/textual_inversion.py
+1 -2
View File
@@ -430,8 +430,7 @@ def train_embedding(id_task, embedding_name, learn_rate, batch_size, gradient_st
shared.log.info("No saved optimizer exists in checkpoint")
if shared.cmd_opts.use_ipex:
shared.sd_model.train()
shared.sd_model, optimizer = torch.xpu.optimize(shared.sd_model.to(dtype=torch.float32), optimizer=optimizer, dtype=devices.dtype)
pass
else:
scaler = torch.cuda.amp.GradScaler()