Fixed Major bug (#113)

easy-model-fusion · Apr 15, 2024 · c52c7c8 · c52c7c8
1 parent 26e7136
commit c52c7c8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/sdk/downloader.py b/sdk/downloader.py
@@ -593,7 +593,7 @@ def parse_arguments():
                         help="Model name")
 
     # Optional arguments regarding the model
-    parser.add_argument("--model_module", type=str,
+    parser.add_argument("--model-module", type=str,
                         help="Module name", choices=AUTHORIZED_MODULE_NAMES)
     parser.add_argument("--model-class", type=str,
                         help="Class name within the module")