GC

app.py

@@ -4,6 +4,7 @@ import random
 import gradio as gr
 from diffusers import AutoPipelineForText2Image
 from diffusers.utils import load_image
+import gc
 
 device = "cuda" if torch.cuda.is_available() else "cpu"
 dtype = torch.float16 if torch.cuda.is_available() else torch.float32
@@ -56,10 +57,14 @@ def create_image(image_pil,
         generator=generator,
     ).images[0]
     
+    if torch.cuda.is_available():
+        torch.cuda.empty_cache()
+        gc.collect()
     
     return image
     
     
+    
 
 # Description
 title = r"""