Hi, this is an official implementation of StyleCrafter in SDXL We train StyleCrafter on SDXL to further enhance its generated quality for style-guided image generation.
TL;DR: Higher Resolution(1024×1024)! More Visually Pleasing!
Style-guided text-to-image results. Resolution: 1024 x 1024. (Compressed)
conda create -n style_crafter python=3.9
conda activate style_crafter
conda install cudatoolkit=11.8 cudnn
pip3 install torch torchvision torchaudio --index-url https://download.pytorch.org/whl/cu118
pip install diffusers==0.29.1
pip install accelerate==0.31.0
pip install transformers tensorboard omegaconf opencv-python webdatasetDownload StyleCrafter-SDXL checkpoints from huggingface, and put them into the folder ./pretrained_ckpts/.
After downloading and moving, the directiry structure should look like this:
pretrained_ckpts
├── image_encoder
│ ├── config.json
│ └── pytorch_model.bin
└── stylecrafter
└── stylecrafter_sdxl.ckpt
Run the following command to generate stylized videos.
python infer.py --style_dir testing_data/input_style \
--prompts_file testing_data/prompts.txt \
--save_dir testing_data/output \
--scale 0.5
If you find unsatisfactory results, try slightly adjusting the scale value. Empirically, reduce the scale if it produces artifacts, and increase the scale if result is less stylized.
-
Prepare your own training data as webdataset style, or just modified dataset.py to adapted to your data as preferred.
-
launch the training bash(based on accelerate)
sh train.shAs a reference, we train StyleCrafter-SDXL as the following steps:
- Train at resolution 512×512 for 80k steps, with batchsize=128, lr=5e-5, no noise offset;
- Train at resolution 1024×1024 for 80k steps, with batchsize=64, lr=2e-5, no noise offset;
- Train at resolution 1024×1024 for 40k steps, with batchsize=64, lr=1e-5, noise_offset=0.05;
We conduct all the training processes on 8 Nvidia A100 GPUs, which takes about a week to complete. Just approximation.
For more details(model arch, data process, etc.), please refer to our paper:
StyleCrafter: Enhancing Stylized Text-to-Video Generation with Style Adapter
GongyeLiu,
Menghan Xia*,
Yong Zhang,
Haoxin Chen,
Jinbo Xing,
Xintao Wang,
Ying Shan
Yujiu Yang*
(* corresponding authors)
StyleCrafter Github Repo(based on VideoCrafter)
We develop this repository for RESEARCH purposes, so it can only be used for personal/research/non-commercial purposes.
This repo is based on diffusers and accelerate, and our training code for SDXL is largely modified from IP-Adapter. We would like to thank them for their awesome contributions to the AIGC community.
If your have any comments or questions, feel free to contact [email protected]
@article{liu2023stylecrafter,
title={StyleCrafter: Enhancing Stylized Text-to-Video Generation with Style Adapter},
author={Liu, Gongye and Xia, Menghan and Zhang, Yong and Chen, Haoxin and Xing, Jinbo and Wang, Xintao and Yang, Yujiu and Shan, Ying},
journal={arXiv preprint arXiv:2312.00330},
year={2023}
}