Update app.py

app.py

@@ -124,10 +124,16 @@ class MagicTimeController:
         converted_vae_checkpoint = convert_ldm_vae_checkpoint(dreambooth_state_dict, self.vae.config)
         self.vae.load_state_dict(converted_vae_checkpoint)
 
+        if self.unet is not None:
+            del self.unet  # 删除旧模型的引用
+            torch.cuda.empty_cache()
         converted_unet_checkpoint = convert_ldm_unet_checkpoint(dreambooth_state_dict, self.unet_model.config)
         self.unet = copy.deepcopy(self.unet_model)
         self.unet.load_state_dict(converted_unet_checkpoint, strict=False)
 
+        if self.text_encoder is not None:
+            del self.text_encoder
+            torch.cuda.empty_cache()
         text_model = copy.deepcopy(self.text_model)
         self.text_encoder = convert_ldm_clip_text_model(text_model, dreambooth_state_dict)