From 27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sat, 24 Jun 2023 16:26:22 +0200 Subject: Fixes --- training/strategy/dreambooth.py | 3 --- 1 file changed, 3 deletions(-) (limited to 'training/strategy') 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( unet_ = accelerator.unwrap_model(unet, keep_fp32_wrapper=False) text_encoder_ = accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False) - unet_.forward = MethodType(unet_.forward, unet_) - text_encoder_.forward = MethodType(text_encoder_.forward, text_encoder_) - with ema_context(): pipeline = VlpnStableDiffusion( text_encoder=text_encoder_, -- cgit v1.2.3-54-g00ecf