mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
4fa24591a3
* create a script to train vae * update main.py * update train_autoencoderkl.py * update train_autoencoderkl.py * add a check of --pretrained_model_name_or_path and --model_config_name_or_path * remove the comment, remove diffusers in requiremnets.txt, add validation_image ote * update autoencoderkl.py * quality --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
16 lines
157 B
Plaintext
16 lines
157 B
Plaintext
accelerate>=0.16.0
|
|
bitsandbytes
|
|
datasets
|
|
huggingface_hub
|
|
lpips
|
|
numpy
|
|
packaging
|
|
Pillow
|
|
taming_transformers
|
|
torch
|
|
torchvision
|
|
tqdm
|
|
transformers
|
|
wandb
|
|
xformers
|