EngineX-Iluvatar/enginex-mr_series-text2video
15
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

support MODEL_TYPE from ENV

Browse Source
This commit is contained in:
Zhang Hao
2025-09-09 14:22:25 +08:00
parent 4525010edc
commit 54ac219d77
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
main.py
View File

@@ -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)