RealVis_v5.0_BF16_B

Running on Zero

ford442 commited on Dec 25, 2024

Commit

0eece6c

verified ·

1 Parent(s): 400f932

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -278,8 +278,8 @@ def generate_30(
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
-    torch.cuda.empty_cache()
-    gc.collect()
     return sd_image_path, seed
 @spaces.GPU(duration=60)
@@ -335,8 +335,8 @@ def generate_60(
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
-    torch.cuda.empty_cache()
-    gc.collect()
     return sd_image_path, seed
 @spaces.GPU(duration=90)
@@ -392,8 +392,8 @@ def generate_90(
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
-    torch.cuda.empty_cache()
-    gc.collect()
     return sd_image_path, seed
 def load_predefined_images1():

     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
+    #torch.cuda.empty_cache()
+    #gc.collect()
     return sd_image_path, seed
 @spaces.GPU(duration=60)
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
+    #torch.cuda.empty_cache()
+    #gc.collect()
     return sd_image_path, seed
 @spaces.GPU(duration=90)
     rv_image.save(sd_image_path,optimize=False,compress_level=0)
     upload_to_ftp(sd_image_path)
     #image_paths = save_image(rv_image)
+    #torch.cuda.empty_cache()
+    #gc.collect()
     return sd_image_path, seed
 def load_predefined_images1():