Uniaff commited on
Commit
7ad861e
·
verified ·
1 Parent(s): c3dd560

Update inference.py

Browse files
Files changed (1) hide show
  1. inference.py +4 -3
inference.py CHANGED
@@ -46,9 +46,10 @@ def parse_arguments():
46
  parser.add_argument('--pads', nargs=4, type=int, default=[0, 10, 0, 0],
47
  help='Padding (top, bottom, left, right). Please adjust to include chin at least')
48
 
49
- parser.add_argument('--face_det_batch_size', type=int,
50
- help='Batch size for face detection', default=16)
51
- parser.add_argument('--wav2lip_batch_size', type=int, help='Batch size for Wav2Lip model(s)', default=128)
 
52
 
53
  parser.add_argument('--resize_factor', default=1, type=int,
54
  help='Reduce the resolution by this factor. Sometimes, best results are obtained at 480p or 720p')
 
46
  parser.add_argument('--pads', nargs=4, type=int, default=[0, 10, 0, 0],
47
  help='Padding (top, bottom, left, right). Please adjust to include chin at least')
48
 
49
+ parser.add_argument('--face_det_batch_size', type=int, help='Batch size for face detection', default=32)
50
+ parser.add_argument('--wav2lip_batch_size', type=int, help='Batch size for Wav2Lip model(s)', default=256)
51
+
52
+
53
 
54
  parser.add_argument('--resize_factor', default=1, type=int,
55
  help='Reduce the resolution by this factor. Sometimes, best results are obtained at 480p or 720p')