Spaces:
Paused
Paused
update.
Browse files- inference_propainter.py +3 -2
inference_propainter.py
CHANGED
|
@@ -217,8 +217,9 @@ if __name__ == '__main__':
|
|
| 217 |
args = parser.parse_args()
|
| 218 |
|
| 219 |
# Use fp16 precision during inference to reduce running memory cost
|
| 220 |
-
use_half = True if args.fp16 else False
|
| 221 |
-
|
|
|
|
| 222 |
|
| 223 |
frames, fps, size, video_name = read_frame_from_videos(args.video)
|
| 224 |
if not args.width == -1 and not args.height == -1:
|
|
|
|
| 217 |
args = parser.parse_args()
|
| 218 |
|
| 219 |
# Use fp16 precision during inference to reduce running memory cost
|
| 220 |
+
use_half = True if args.fp16 else False
|
| 221 |
+
if device == torch.device('cpu'):
|
| 222 |
+
use_half = False
|
| 223 |
|
| 224 |
frames, fps, size, video_name = read_frame_from_videos(args.video)
|
| 225 |
if not args.width == -1 and not args.height == -1:
|