diff --git a/.gitignore b/.gitignore index deeef7a..a112fdb 100644 --- a/.gitignore +++ b/.gitignore @@ -34,4 +34,6 @@ Wan2.1-T2V-14B/ Wan2.1-T2V-1.3B/ Wan2.1-I2V-14B-480P/ Wan2.1-I2V-14B-720P/ -poetry.lock \ No newline at end of file +poetry.lock +wok/37ec512624d61f7aa208f7ea8140a131f93afc9a +wok/t2v-1.3b \ No newline at end of file diff --git a/environment.yml b/environment.yml new file mode 100644 index 0000000..43aaa7e --- /dev/null +++ b/environment.yml @@ -0,0 +1,24 @@ +name: wan21 +channels: + - conda-forge + - defaults +dependencies: + - python>=3.10 + - pytorch>=2.4.0 + - torchvision>=0.19.0 + - tqdm + - imageio + - imageio-ffmpeg + - numpy>=1.23.5,<2 + - pip + - pip: + - opencv-python>=4.9.0.80 + - diffusers>=0.31.0 + - transformers>=4.49.0 + - tokenizers>=0.20.3 + - accelerate>=1.1.1 + - easydict + - ftfy + - dashscope + - flash_attn + - gradio>=5.0.0 diff --git a/wok/go.sh b/wok/go.sh new file mode 100755 index 0000000..e911a74 --- /dev/null +++ b/wok/go.sh @@ -0,0 +1,2 @@ +#!/usr/bin/bash +python ../generate.py --task t2v-1.3B --size 480*832 --ckpt_dir ./t2v-1.3b --offload_model True --prompt "happy the dwarf and sneezy the dwarf wrestle to the death at madison square garden"