CoolFace
Modelpublic

Alibaba-Research-Intelligence-Computing/wan-toy-transform

sourceHugging Facemitupdated 1y agoView on Hugging Face
6likes11downloads
Model Card

Please refer to our github for more info: https://github.com/alibaba/wan-toy-transform

<div align="center"> <h2><center>Wan Toy Transform</h2> <br> Alibaba Research Intelligence Computing <br> <a href="https://github.com/alibaba/wan-toy-transform"><img src='https://img.shields.io/badge/Github-Link-black'></a> <a href='https://modelscope.cn/models/AlibabaResearchIntelligenceComputing/wan-toy-transform'><img src='https://img.shields.io/badge/๐Ÿค–ModelScope-weights-%23654dfc'></a> <a href='https://huggingface.co/Alibaba-Research-Intelligence-Computing/wan-toy-transform'><img src='https://img.shields.io/badge/๐Ÿค—_HuggingFace-weights-%23ff9e0e'></a> <br> </div>

This is a LoRA model finetuned on Wan-I2V-14B-480P. It turns things in the image into fluffy toys.

๐Ÿ Installation

bash
# Python 3.12 and PyTorch 2.6.0 are tested.
pip install torch==2.6.0 torchvision==0.21.0 --index-url https://download.pytorch.org/whl/cu124
pip install -r requirements.txt

๐Ÿ”„ Inference

bash
python generate.py --prompt "The video opens with a clear view of a $name. Then it transforms to a b6e9636 JellyCat-style $name. It has a face and a cute, fluffy and playful appearance." --image $image_path --save_file "output.mp4" --offload_type leaf_level

Note:

  • โ€”Change $name to the object name you want to transform.
  • โ€”$image_path is the path to the first frame image.
  • โ€”Choose --offload_type from ['leaflevel', 'blocklevel', 'none', 'model']. More details can be found here.
  • โ€”VRAM usage and generation time of different --offload_type are listed below.
`--offload_type`VRAM UsageGeneration Time (NVIDIA A100)
leaf_level11.9 GB17m17s
blocklevel (numblockspergroup=1)20.5 GB16m48s
model39.4 GB16m24s
none55.9 GB16m08s

๐Ÿค Acknowledgements

Special thanks to these projects for their contributions to the community!

๐Ÿ“„ Our previous work