Spaces:
Sleeping
Sleeping
Commit
·
cff8dcb
1
Parent(s):
2f7bc8e
back to cuda
Browse files
app.py
CHANGED
|
@@ -99,8 +99,7 @@ def upscale2(image, prompt):
|
|
| 99 |
print("upscale2",image,prompt)
|
| 100 |
|
| 101 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
-
|
| 103 |
-
pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages")
|
| 104 |
pipeline = pipeline.to(device)
|
| 105 |
|
| 106 |
upscaled_image = pipeline(image, num_inference_steps=10, eta=1).images[0]
|
|
|
|
| 99 |
print("upscale2",image,prompt)
|
| 100 |
|
| 101 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
+
pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages", torch_dtype=torch.float16)
|
|
|
|
| 103 |
pipeline = pipeline.to(device)
|
| 104 |
|
| 105 |
upscaled_image = pipeline(image, num_inference_steps=10, eta=1).images[0]
|