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:
|