Wan2.1 项目使用与配置指南
1. 项目目录结构及介绍
Wan2.1 项目目录结构如下:
Wan-Video/
├── assets/ # 存放项目相关的资源文件
├── examples/ # 包含项目使用示例
├── gradio/ # Gradio 相关的示例和界面
├── tests/ # 测试代码目录
├── wan/ # 核心代码目录
├── .gitignore # 指定 Git 忽略的文件
├── INSTALL.md # 安装指南
├── LICENSE.txt # 开源协议文件
├── README.md # 项目说明文件
├── generate.py # 视频生成脚本
├── pyproject.toml # 项目配置文件
└── requirements.txt # 项目依赖文件
assets/
:包含项目所需的各种资源文件,如图片、视频等。examples/
:提供了一些项目使用示例,方便用户快速入门。gradio/
:用于创建 Gradio 界面,便于展示和测试模型效果。tests/
:包含了项目的单元测试代码,用于确保代码质量。wan/
:包含了项目的主要代码,包括模型定义、数据处理等。
2. 项目的启动文件介绍
项目的启动文件是 generate.py
,该脚本用于启动视频生成流程。以下是一个简单的启动示例:
python generate.py --task t2v-14B --size 1280x720 --ckpt_dir ./Wan2.1-T2V-14B --prompt "Two anthropomorphic cats in comfy boxing gear and bright gloves fight intensely on a spotlighted stage."
其中:
--task
:指定使用的任务类型,如t2v-14B
表示使用文本到视频的任务。--size
:指定生成的视频分辨率。--ckpt_dir
:指定模型权重的存储路径。--prompt
:为生成视频提供的文本描述。
3. 项目的配置文件介绍
项目的配置文件是 pyproject.toml
,该文件包含了项目的基本配置信息。以下是一些常用的配置项:
[tool.poetry]
name = "Wan2.1"
version = "0.1.0"
description = "An open-source video generative model suite."
authors = ["Your Name <your_email@example.com>"]
[tool.poetry.dependencies]
python = "^3.8"
torch = "^2.4.0"
[tool.poetry.dev-dependencies]
pytest = "^6.2.4"
name
:项目的名称。version
:项目的版本号。description
:项目的描述信息。authors
:项目维护者的名字和联系方式。dependencies
:项目运行所依赖的库及其版本。dev-dependencies
:项目开发过程中依赖的库及其版本。
通过配置这些信息,可以方便地管理项目依赖和版本。