EngineX-Ascend/enginex-ascend-910-text2video
10
0
Fork 0
Code Issues Pull Requests Actions Projects Releases Wiki Activity

fix parser

Browse Source
This commit is contained in:
ZHANG Hao
2025-09-03 16:04:18 +08:00
parent eeda961724
commit 96ad5735ac
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
main.py
View File

@@ -97,7 +97,7 @@ def main():
parser.add_argument("--outdir", required=True, help="图片输出目录") parser.add_argument("--outdir", required=True, help="图片输出目录")
parser.add_argument("--device", default="cuda", choices=["cuda", "cpu"], help="推理设备") parser.add_argument("--device", default="cuda", choices=["cuda", "cpu"], help="推理设备")
parser.add_argument("--dtype", default="fp16", choices=["fp16", "fp32"], help="推理精度") parser.add_argument("--dtype", default="fp16", choices=["fp16", "fp32"], help="推理精度")
args, _ = p.parse_known_args() args, _ = parser.parse_known_args()
model_path = args.model model_path = args.model
json_path = Path(args.json) json_path = Path(args.json)