From 54ac219d773fe40bed68cc2ec49d12093cd6b36b Mon Sep 17 00:00:00 2001 From: Zhang Hao Date: Tue, 9 Sep 2025 14:22:25 +0800 Subject: [PATCH] support MODEL_TYPE from ENV --- main.py | 3 ++- 1 file changed, 2 insertions(+), 1 deletion(-) diff --git a/main.py b/main.py index 092d5b0..2116f4b 100644 --- a/main.py +++ b/main.py @@ -103,8 +103,9 @@ def main(): parser.add_argument("--outdir", required=True, help="图片输出目录") parser.add_argument("--device", default="cuda", help="推理设备") parser.add_argument("--dtype", default="fp16", choices=["fp16", "fp32"], help="推理精度") - parser.add_argument("--model_type", default="text-to-video-synthesis", choices=["text-to-video-synthesis", "text-to-video-ms"], help="模型类型") + parser.add_argument("--model_type", default=os.getenv("MODEL_TYPE", "text-to-video-synthesis"), choices=["text-to-video-synthesis", "text-to-video-ms"], help="模型类型") args, _ = parser.parse_known_args() + print("参数:", args) model_path = args.model json_path = Path(args.json)