Step-Video-T2V 项目使用说明

Step-Video-T2V 项目使用说明

Step-Video-T2V 项目地址: https://gitcode.com/gh_mirrors/st/Step-Video-T2V

1. 项目目录结构及介绍

Step-Video-T2V 项目目录结构如下：

Step-Video-T2V/
├── api/
│   ├── call_remote_server.py
├── assets/
├── benchmark/
├── stepvideo/
│   ├── __init__.py
│   ├── model.py
│   ├── modules.py
│   └── ...
├── .gitignore
├── LICENSE
├── README.md
├── run_parallel.py
├── setup.py
└── ...

• api/: 包含与远程服务器通信的脚本，例如 call_remote_server.py。
• assets/: 存储项目所需的静态资源。
• benchmark/: 包含性能测试相关的代码和数据。
• stepvideo/: 核心代码目录，包含模型定义、模块等。
• .gitignore: 指定 Git 忽略的文件和目录。
• LICENSE: 项目使用的许可文件。
• README.md: 项目的说明文档。
• run_parallel.py: 并行运行的脚本。
• setup.py: 项目安装和依赖配置的脚本。

2. 项目的启动文件介绍

项目的启动文件为 run_parallel.py，该脚本负责启动多 GPU 并行训练或推理过程。以下是一个简单的启动命令示例：

torchrun --nproc_per_node <parallel_gpu_count> run_parallel.py --model_dir <path_to_model_directory> --vae_url <vae_api_url> --caption_url <caption_api_url> --ulysses_degree <degree> --tensor_parallel_degree <degree> --prompt "<your_text_prompt>" --infer_steps <steps> --cfg_scale <scale> --time_shift <shift>

其中：

• <parallel_gpu_count>: 并行使用的 GPU 数量。
• <path_to_model_directory>: 模型文件所在的目录。
• <vae_api_url>: VAE 解码服务的 URL。
• <caption_api_url>: 文本编码服务的 URL。
• <degree>: 用于并行计算的度数。
• <your_text_prompt>: 用户提供的文本提示。
• <steps>, <scale>, <shift>: 推理时的超参数设置。

3. 项目的配置文件介绍

项目的配置文件通常在 setup.py 中定义，它包含了项目的基本信息和依赖。以下是一个配置文件的示例：

from setuptools import setup, find_packages

setup(
    name='Step-Video-T2V',
    version='0.1.0',
    packages=find_packages(),
    install_requires=[
        'torch>=2.3-cu121',
        # 其他依赖
    ]
)

在这个配置文件中，定义了项目的名称、版本号、包含的包以及安装时需要依赖的其他 Python 包。这些依赖在安装项目时会自动处理。

请注意，实际项目可能包含更多的配置细节和选项，上述内容仅为示例。

Step-Video-T2V 项目地址: https://gitcode.com/gh_mirrors/st/Step-Video-T2V