Spaces:
Sleeping
Sleeping
Commit
·
f54cef3
1
Parent(s):
656c364
fix
Browse files
app.py
CHANGED
@@ -71,7 +71,7 @@ def segmentation(img):
|
|
71 |
def upscale(image,prompt):
|
72 |
print("upscale")
|
73 |
|
74 |
-
image =
|
75 |
|
76 |
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
|
77 |
# pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
|
|
|
71 |
def upscale(image,prompt):
|
72 |
print("upscale")
|
73 |
|
74 |
+
image = image.thumbnail((512, 512))
|
75 |
|
76 |
pipeline = DiffusionPipeline.from_pretrained("stabilityai/stable-diffusion-x4-upscaler")
|
77 |
# pipeline = StableDiffusionLatentUpscalePipeline.from_pretrained("stabilityai/sd-x2-latent-upscaler", torch_dtype=torch.float16).to('cuda')
|