inf2_env.py wrapper: env var consistency fix

oOraph · oOraph · commit c37dc1d542d5 · 2025-04-01T20:07:42.000+02:00
use the same HF_MODEL_ID env var as the main server

Signed-off-by: Raphael Glon &lt;oOraph@users.noreply.github.com&gt;
diff --git a/scripts/inf2_env.py b/scripts/inf2_env.py
@@ -50,7 +50,7 @@ def parse_cmdline_and_set_env(argv: List[str] = None) -> argparse.Namespace:
                           os.getenv("SEQUENCE_LENGTH", 0))
     )
 
-    parser.add_argument("--model-id", type=str, default=os.getenv("MODEL_ID", os.getenv("HF_MODEL_DIR")))
+    parser.add_argument("--model-id", type=str, default=os.getenv("HF_MODEL_ID", os.getenv("HF_MODEL_DIR")))
     parser.add_argument("--revision", type=str, default=os.getenv("REVISION"))
 
     args = parser.parse_known_args(argv)[0]

Original file line number	Diff line number	Diff line change
`@@ -50,7 +50,7 @@ def parse_cmdline_and_set_env(argv: List[str] = None) -> argparse.Namespace:`
`50`	`50`	`os.getenv("SEQUENCE_LENGTH", 0))`
`51`	`51`	`)`
`52`	`52`
`53`		`- parser.add_argument("--model-id", type=str, default=os.getenv("MODEL_ID", os.getenv("HF_MODEL_DIR")))`
	`53`	`+ parser.add_argument("--model-id", type=str, default=os.getenv("HF_MODEL_ID", os.getenv("HF_MODEL_DIR")))`
`54`	`54`	`parser.add_argument("--revision", type=str, default=os.getenv("REVISION"))`
`55`	`55`
`56`	`56`	`args = parser.parse_known_args(argv)[0]`