diff options
| author | Volpeon <git@volpeon.ink> | 2023-06-24 16:26:22 +0200 |
|---|---|---|
| committer | Volpeon <git@volpeon.ink> | 2023-06-24 16:26:22 +0200 |
| commit | 27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712 (patch) | |
| tree | 6c1f2243475778bb5e9e1725bf3969a5442393d8 /training/strategy | |
| parent | Update (diff) | |
| download | textual-inversion-diff-27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712.tar.gz textual-inversion-diff-27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712.tar.bz2 textual-inversion-diff-27b18776ba6d38d6bda5e5bafee3e7c4ca8c9712.zip | |
Fixes
Diffstat (limited to 'training/strategy')
| -rw-r--r-- | training/strategy/dreambooth.py | 3 |
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_, |