Spaces:
Runtime error
Runtime error
Update app.py
Browse files
app.py
CHANGED
|
@@ -144,6 +144,10 @@ def get_image(image, sketch, prompt, steps=50, cfg_scale=7.5, eta=1.0, fs=3, see
|
|
| 144 |
transforms.Resize(min(resolution)),
|
| 145 |
transforms.CenterCrop(resolution),
|
| 146 |
])
|
|
|
|
|
|
|
|
|
|
|
|
|
| 147 |
print("before empty cache")
|
| 148 |
torch.cuda.empty_cache()
|
| 149 |
print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
|
|
|
|
| 144 |
transforms.Resize(min(resolution)),
|
| 145 |
transforms.CenterCrop(resolution),
|
| 146 |
])
|
| 147 |
+
|
| 148 |
+
transform = transforms.Compose([
|
| 149 |
+
transforms.Resize(resolution),
|
| 150 |
+
])
|
| 151 |
print("before empty cache")
|
| 152 |
torch.cuda.empty_cache()
|
| 153 |
print('start:', prompt, time.strftime('%Y-%m-%d %H:%M:%S',time.localtime(time.time())))
|