sczhou commited on
Commit
87ca85b
β€’
1 Parent(s): 320e465
Files changed (1) hide show
  1. 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: