StableDiffusion-3.5-Large-lora-test

Running on Zero

1inkusFace commited on May 13

Commit

3190216

verified ·

1 Parent(s): af8fe0c

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -150,7 +150,7 @@ def infer_60(
     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
-    upscaler_2.to(torch.device('cuda',non_blocking=True))
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
@@ -200,7 +200,7 @@ def infer_90(
     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
-    upscaler_2.to(torch.device('cuda',non_blocking=True))
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
@@ -250,7 +250,7 @@ def infer_110(
     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
-    upscaler_2.to(torch.device('cuda',non_blocking=True))
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)

     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
+    upscaler_2.to(torch.device('cuda'),non_blocking=True)
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
+    upscaler_2.to(torch.device('cuda'),non_blocking=True)
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)
     sd35_path = f"sd35ll_{timestamp}.png"
     sd_image.save(sd35_path,optimize=False,compress_level=0)
     pyx.upload_to_ftp(sd35_path)
+    upscaler_2.to(torch.device('cuda'),non_blocking=True)
     with torch.no_grad():
         upscale = upscaler_2(sd_image, tiling=True, tile_width=256, tile_height=256)
         upscale2 = upscaler_2(upscale, tiling=True, tile_width=256, tile_height=256)