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')
|