zzl
commited on
Commit
•
b5a9347
1
Parent(s):
4fa279e
release
Browse files- demo_img.py +1 -1
- demo_vid.py +1 -1
demo_img.py
CHANGED
|
@@ -34,7 +34,7 @@ def img2vid(model_type, img0, img1, frame_ratio, iters):
|
|
| 34 |
anchor_memory = 1500 * 1024**2
|
| 35 |
anchor_memory_bias = 2500 * 1024**2
|
| 36 |
vram_avail = torch.cuda.get_device_properties(device).total_memory
|
| 37 |
-
else
|
| 38 |
# Do not resize in cpu mode
|
| 39 |
anchor_resolution = 8192*8192
|
| 40 |
anchor_memory = 1
|
|
|
|
| 34 |
anchor_memory = 1500 * 1024**2
|
| 35 |
anchor_memory_bias = 2500 * 1024**2
|
| 36 |
vram_avail = torch.cuda.get_device_properties(device).total_memory
|
| 37 |
+
else:
|
| 38 |
# Do not resize in cpu mode
|
| 39 |
anchor_resolution = 8192*8192
|
| 40 |
anchor_memory = 1
|
demo_vid.py
CHANGED
|
@@ -32,7 +32,7 @@ def vid2vid(model_type, video, iters):
|
|
| 32 |
anchor_memory = 1500 * 1024**2
|
| 33 |
anchor_memory_bias = 2500 * 1024**2
|
| 34 |
vram_avail = torch.cuda.get_device_properties(device).total_memory
|
| 35 |
-
else
|
| 36 |
# Do not resize in cpu mode
|
| 37 |
anchor_resolution = 8192*8192
|
| 38 |
anchor_memory = 1
|
|
|
|
| 32 |
anchor_memory = 1500 * 1024**2
|
| 33 |
anchor_memory_bias = 2500 * 1024**2
|
| 34 |
vram_avail = torch.cuda.get_device_properties(device).total_memory
|
| 35 |
+
else:
|
| 36 |
# Do not resize in cpu mode
|
| 37 |
anchor_resolution = 8192*8192
|
| 38 |
anchor_memory = 1
|