summaryrefslogtreecommitdiffstats
path: root/training/strategy
diff options
context:
space:
mode:
Diffstat (limited to 'training/strategy')
-rw-r--r--training/strategy/dreambooth.py3
1 files changed, 0 insertions, 3 deletions
diff --git a/training/strategy/dreambooth.py b/training/strategy/dreambooth.py
index 0f64747..bd853e2 100644
--- a/training/strategy/dreambooth.py
+++ b/training/strategy/dreambooth.py
@@ -155,9 +155,6 @@ def dreambooth_strategy_callbacks(
155 unet_ = accelerator.unwrap_model(unet, keep_fp32_wrapper=False) 155 unet_ = accelerator.unwrap_model(unet, keep_fp32_wrapper=False)
156 text_encoder_ = accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False) 156 text_encoder_ = accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False)
157 157
158 unet_.forward = MethodType(unet_.forward, unet_)
159 text_encoder_.forward = MethodType(text_encoder_.forward, text_encoder_)
160
161 with ema_context(): 158 with ema_context():
162 pipeline = VlpnStableDiffusion( 159 pipeline = VlpnStableDiffusion(
163 text_encoder=text_encoder_, 160 text_encoder=text_encoder_,