Update app.py

app.py

@@ -333,7 +333,7 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
                 return image
         else:
             if upscale == "Yes":
-                
+                torch.cuda.empty_cache()
                 image = pipe(Prompt, negative_prompt=negative_prompt, height=height, width=width, num_inference_steps=steps, guidance_scale=scale).images[0]
                 torch.cuda.empty_cache()
                 torch.cuda.max_memory_allocated(device=device)
@@ -351,9 +351,12 @@ def genie (Model, Prompt, negative_prompt, height, width, scale, steps, seed, re
                return image
 
     if Model == 'Cascade':
+        torch.cuda.empty_cache()
         from diffusers import StableCascadeCombinedPipeline
         pipe = StableCascadeCombinedPipeline.from_pretrained("stabilityai/stable-cascade", variant="bf16", torch_dtype=torch.bfloat16)
-        image = (prompt=Prompt, negative_prompt=negative_prompt, num_inference_steps=10, prior_num_inference_steps=20, prior_guidance_scale=3.0, width=width, height=height).images[0]
+        pipe.enable_xformers_memory_efficient_attention()
+        pipe = pipe.to(device)
+        image = pipe(prompt=Prompt, negative_prompt=negative_prompt, num_inference_steps=10, prior_num_inference_steps=20, prior_guidance_scale=3.0, width=width, height=height).images[0]
         return image
 
     return image