fix
Browse files
app.py
CHANGED
|
@@ -88,7 +88,8 @@ def sepia(image_web, seed):
|
|
| 88 |
output_image = model.web_restore(image, r=config.data.grid_r)
|
| 89 |
grid = make_grid(output_image)
|
| 90 |
ndarr = grid.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to("cpu", torch.uint8).numpy()
|
| 91 |
-
torch.cuda.empty_cache()
|
|
|
|
| 92 |
return ndarr, image_tmp
|
| 93 |
|
| 94 |
|
|
|
|
| 88 |
output_image = model.web_restore(image, r=config.data.grid_r)
|
| 89 |
grid = make_grid(output_image)
|
| 90 |
ndarr = grid.mul(255).add_(0.5).clamp_(0, 255).permute(1, 2, 0).to("cpu", torch.uint8).numpy()
|
| 91 |
+
# torch.cuda.empty_cache()
|
| 92 |
+
print("生成完成")
|
| 93 |
return ndarr, image_tmp
|
| 94 |
|
| 95 |
|