Update app.py

app.py

@@ -101,7 +101,7 @@ generator = torch.Generator(device='cuda:0').manual_seed(123456)
 
 vae = pipe.vae
 
-
+pipe.enable_model_cpu_offload() 
 # pipe.force_zeros_for_empty_prompt = False
 
 # default_negative_prompt= "" #"Logo,Watermark,Text,Ugly,Morbid,Extra fingers,Poorly drawn hands,Mutation,Blurry,Extra limbs,Gross proportions,Missing arms,Mutated hands,Long neck,Duplicate,Mutilated,Mutilated hands,Poorly drawn face,Deformed,Bad anatomy,Cloned face,Malformed limbs,Missing legs,Too many fingers"
@@ -183,7 +183,7 @@ def predict(dict, prompt="", negative_prompt = default_negative_prompt, guidance
     #         guidance_scale = 1.2,
     #         generator=generator).images[0]
 
-    
+    torch.cuda.empty_cache
     return output.images[0] #, gr.update(visible=True)