Spaces:
Running
on
Zero
Running
on
Zero
lzyhha
commited on
Commit
·
5e74212
1
Parent(s):
ccc05b7
cache
Browse files- visualcloze.py +2 -0
visualcloze.py
CHANGED
|
@@ -430,6 +430,8 @@ class VisualClozeModel:
|
|
| 430 |
output_image = to_pil_image(row_sample.float())
|
| 431 |
output_images.append(output_image)
|
| 432 |
|
|
|
|
|
|
|
| 433 |
ret = []
|
| 434 |
ret_w = output_images[-1].width
|
| 435 |
ret_h = output_images[-1].height
|
|
|
|
| 430 |
output_image = to_pil_image(row_sample.float())
|
| 431 |
output_images.append(output_image)
|
| 432 |
|
| 433 |
+
torch.cuda.empty_cache()
|
| 434 |
+
|
| 435 |
ret = []
|
| 436 |
ret_w = output_images[-1].width
|
| 437 |
ret_h = output_images[-1].height
|