up2

app.py

@@ -75,13 +75,14 @@ def upscale(image, prompt):
     image.thumbnail((512, 512))
     print("resize",image)
 
-    pipeline = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", torch_dtype=torch.float16)
-    pipeline = pipeline.to("cuda")
+    # pipeline = StableDiffusionUpscalePipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler", torch_dtype=torch.float16)
+    # pipeline = pipeline.to("cuda")
 
     #up
 
     #pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
-    # pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
+    pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
+    
     ret = pipeline(prompt=prompt, 
                    image=image,
                    num_inference_steps=20,