diff options
author | Volpeon <git@volpeon.ink> | 2023-06-25 08:40:05 +0200 |
---|---|---|
committer | Volpeon <git@volpeon.ink> | 2023-06-25 08:40:05 +0200 |
commit | 4f6ecc2fd38dc894ac7fcb4f130d3ab8af49d132 (patch) | |
tree | c6ab59d6726b818638fe90a3ea8bb8403a8b0c30 /training/functional.py | |
parent | Update (diff) | |
download | textual-inversion-diff-4f6ecc2fd38dc894ac7fcb4f130d3ab8af49d132.tar.gz textual-inversion-diff-4f6ecc2fd38dc894ac7fcb4f130d3ab8af49d132.tar.bz2 textual-inversion-diff-4f6ecc2fd38dc894ac7fcb4f130d3ab8af49d132.zip |
Update
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() |