Spaces:
Sleeping
Sleeping
Commit
·
2f7bc8e
1
Parent(s):
11d7771
cpu
Browse files
app.py
CHANGED
|
@@ -99,6 +99,7 @@ def upscale2(image, prompt):
|
|
| 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")
|
| 103 |
pipeline = pipeline.to(device)
|
| 104 |
|
|
|
|
| 99 |
print("upscale2",image,prompt)
|
| 100 |
|
| 101 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
| 102 |
+
device = "cpu"
|
| 103 |
pipeline = LDMSuperResolutionPipeline.from_pretrained("CompVis/ldm-super-resolution-4x-openimages")
|
| 104 |
pipeline = pipeline.to(device)
|
| 105 |
|