Spaces:
No application file
No application file
Update infer_cli.py
Browse files- infer_cli.py +4 -4
infer_cli.py
CHANGED
|
@@ -32,11 +32,11 @@ def main():
|
|
| 32 |
load_dotenv()
|
| 33 |
args = arg_parse()
|
| 34 |
|
| 35 |
-
|
| 36 |
-
|
| 37 |
-
|
| 38 |
|
| 39 |
-
vc = VC()
|
| 40 |
|
| 41 |
# Add a check for the model name
|
| 42 |
if not args.model_name:
|
|
|
|
| 32 |
load_dotenv()
|
| 33 |
args = arg_parse()
|
| 34 |
|
| 35 |
+
config = Config()
|
| 36 |
+
config.device = args.device if args.device else config.device
|
| 37 |
+
config.is_half = args.is_half if args.is_half else config.is_half
|
| 38 |
|
| 39 |
+
vc = VC(config)
|
| 40 |
|
| 41 |
# Add a check for the model name
|
| 42 |
if not args.model_name:
|