Skip to content

GitLab

Commit 48426398 authored by Xinchi Huang's avatar Xinchi Huang Committed by GitHub
Browse files

async infer.py (#51) 


Co-authored-by: default avatar“de1star” <“843414674@qq.com”>
parent a852f879
Hide whitespace changes
Inline Side-by-side
Showing with 7 additions and 2 deletions
lightx2v/infer.py
View file @ 48426398
import asyncio
import argparse import argparse
import torch import torch
import torch.distributed as dist import torch.distributed as dist
...@@ -33,7 +34,7 @@ def init_runner(config): ...@@ -33,7 +34,7 @@ def init_runner(config):
return runner return runner
if __name__ == "__main__": async def main():
parser = argparse.ArgumentParser() parser = argparse.ArgumentParser()
parser.add_argument("--model_cls", type=str, required=True, choices=["wan2.1", "hunyuan", "wan2.1_causvid", "wan2.1_skyreels_v2_df"], default="hunyuan") parser.add_argument("--model_cls", type=str, required=True, choices=["wan2.1", "hunyuan", "wan2.1_causvid", "wan2.1_skyreels_v2_df"], default="hunyuan")
parser.add_argument("--task", type=str, choices=["t2v", "i2v"], default="t2v") parser.add_argument("--task", type=str, choices=["t2v", "i2v"], default="t2v")
...@@ -54,4 +55,8 @@ if __name__ == "__main__": ...@@ -54,4 +55,8 @@ if __name__ == "__main__":
logger.info(f"config:\n{json.dumps(config, ensure_ascii=False, indent=4)}") logger.info(f"config:\n{json.dumps(config, ensure_ascii=False, indent=4)}")
runner = init_runner(config) runner = init_runner(config)
runner.run_pipeline() await runner.run_pipeline()
if __name__ == "__main__":
asyncio.run(main())
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment