Spaces:
Running
on
A10G
Running
on
A10G
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:
|