From 0d101b9def0f2ea330dbc4899b686813be13aab1 Mon Sep 17 00:00:00 2001 From: Disty0 Date: Tue, 13 Jun 2023 15:36:41 +0300 Subject: [PATCH] Revert xpu.optimize in training --- modules/hypernetworks/hypernetwork.py | 3 +-- modules/textual_inversion/textual_inversion.py | 3 +-- 2 files changed, 2 insertions(+), 4 deletions(-) diff --git a/modules/hypernetworks/hypernetwork.py b/modules/hypernetworks/hypernetwork.py index 15e2c4aaf..c8f1d59da 100644 --- a/modules/hypernetworks/hypernetwork.py +++ b/modules/hypernetworks/hypernetwork.py @@ -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() diff --git a/modules/textual_inversion/textual_inversion.py b/modules/textual_inversion/textual_inversion.py index ba8f429c6..9f9b68c7d 100644 --- a/modules/textual_inversion/textual_inversion.py +++ b/modules/textual_inversion/textual_inversion.py @@ -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()