RealVis_v5.0_BF16_E

Runtime error

ford442 commited on Dec 31, 2024

Commit

02c81e0

verified ·

1 Parent(s): 9e2b698

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -341,8 +341,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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
@@ -403,8 +403,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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
@@ -465,8 +465,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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)

     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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
     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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)
     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()
     torch.set_float32_matmul_precision("medium")
     with torch.no_grad():
         upscale = upscaler(rv_image, tiling=True, tile_width=256, tile_height=256)