1024 * 1024

app.py

@@ -15,7 +15,7 @@ device = "cuda" if torch.cuda.is_available() else "cpu"
 floatType = torch.float16 if torch.cuda.is_available() else torch.float32
 variant = "fp16" if torch.cuda.is_available() else None
 
-pipe = AutoPipelineForImage2Image.from_pretrained("KamCastle/SUPIRcheckpoints", torch_dtype = floatType, variant = variant)
+pipe = AutoPipelineForImage2Image.from_pretrained("stabilityai/sdxl-turbo", torch_dtype = floatType, variant = variant)
 pipe = pipe.to(device)
 
 def update_seed(is_randomize_seed, seed):
@@ -133,6 +133,12 @@ def redraw(
             process_width = process_width - (process_width % 8)
             process_height = process_height - (process_height % 8)
 
+    process_width = 1024
+    process_height = 1024
+
+    output_width = 1024
+    output_height = 1024
+
     progress(None, desc = "Processing...")
     output_image = pipe(
         seeds = [seed],