readme.md 959 Bytes
Newer Older
雍大凯's avatar
雍大凯 committed
1
2
3
4
5
# wan2.1训练

## 环境搭建

```
雍大凯's avatar
readme  
雍大凯 committed
6
docker run -dit --network=host --name=wan21 --privileged --device=/dev/kfd --device=/dev/dri --ipc=host --shm-size=128G  --group-add video --cap-add=SYS_PTRACE --security-opt seccomp=unconfined -u root --ulimit stack=-1:-1 --ulimit memlock=-1:-1 -v /opt/hyhal:/opt/hyhal:ro -v /public/opendas/DL_DATA/llm-models/:/models:ro  harbor.sourcefind.cn:5443/dcu/admin/base/pytorch:2.5.1-ubuntu22.04-dtk25.04.4-1230-py3.10-20260115
雍大凯's avatar
雍大凯 committed
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
cd musubi-tuner
pip install -e .
```

## 数据集下载

```
# 模型可以自行下载开源的
https://huggingface.co/Comfy-Org/Wan_2.1_ComfyUI_repackaged/blob/main/split_files/diffusion_models/wan2.1_i2v_480p_14B_bf16.safetensors
https://huggingface.co/Wan-AI/Wan2.1-I2V-14B-480P

# 数据集下载
链接:https://pan.baidu.com/s/1ehgk0vetX8dj5OEuORZ_jg?pwd=2dsh 
提取码:2dsh 
```

## 执行训练

```
# 自行修改启动脚本内模型及数据集路径
bash start_wan2.1_14b.sh
```