Spaces:

Djrango
/

qwen2vl-flux-mini-demo

Runtime error

erwold commited on Nov 27, 2024

Commit

461d656

1 Parent(s): 7ffc337

ZeroGPU

Files changed (1) hide show

app.py CHANGED Viewed

@@ -270,8 +270,8 @@ def generate_images(input_image, prompt="", guidance_scale=3.5,
             # 将 Transformer 和 VAE 移回 CPU
             logger.info("Moving models back to CPU...")
-            transformer.cpu()
-            vae.cpu()
             torch.cuda.empty_cache()
             return output_images

             # 将 Transformer 和 VAE 移回 CPU
             logger.info("Moving models back to CPU...")
+            #transformer.cpu()
+            #vae.cpu()
             torch.cuda.empty_cache()
             return output_images