diff options
Diffstat (limited to 'training/functional.py')
| -rw-r--r-- | training/functional.py | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/training/functional.py b/training/functional.py index 8917eb7..b60afe3 100644 --- a/training/functional.py +++ b/training/functional.py | |||
| @@ -786,7 +786,4 @@ def train( | |||
| 786 | accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False) | 786 | accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False) |
| 787 | accelerator.unwrap_model(unet, keep_fp32_wrapper=False) | 787 | accelerator.unwrap_model(unet, keep_fp32_wrapper=False) |
| 788 | 788 | ||
| 789 | text_encoder.forward = MethodType(text_encoder.forward, text_encoder) | ||
| 790 | unet.forward = MethodType(unet.forward, unet) | ||
| 791 | |||
| 792 | accelerator.free_memory() | 789 | accelerator.free_memory() |