Spaces:
Running
Running
Update inference.py
Browse files- 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 |
-
|
51 |
-
|
|
|
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')
|