From 4f6ecc2fd38dc894ac7fcb4f130d3ab8af49d132 Mon Sep 17 00:00:00 2001 From: Volpeon Date: Sun, 25 Jun 2023 08:40:05 +0200 Subject: Update --- training/functional.py | 3 --- 1 file changed, 3 deletions(-) (limited to 'training/functional.py') 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( accelerator.unwrap_model(text_encoder, keep_fp32_wrapper=False) accelerator.unwrap_model(unet, keep_fp32_wrapper=False) - text_encoder.forward = MethodType(text_encoder.forward, text_encoder) - unet.forward = MethodType(unet.forward, unet) - accelerator.free_memory() -- cgit v1.2.3-54-g00ecf