RealVis_v5.0_BF16_IP_B

Running on Zero

1inkusFace commited on Apr 20

Commit

1b1775f

verified ·

1 Parent(s): d539a37

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -579,7 +579,7 @@ def generate_30(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
@@ -735,7 +735,7 @@ def generate_60(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
@@ -891,7 +891,7 @@ def generate_90(
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
-        torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)

         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)
         sd_image[0].save(filename,optimize=False,compress_level=0)
         upload_to_ftp(filename)
         uploadNote(prompt,num_inference_steps,guidance_scale,timestamp)
+        #torch.set_float32_matmul_precision("medium")
         with torch.no_grad():
             upscale = upscaler(sd_image, tiling=True, tile_width=256, tile_height=256)
         downscale1 = upscale.resize((upscale.width // 4, upscale.height // 4), Image.LANCZOS)