Bubbliiiingandhkunzhe
9f34f7c8c5
* Update Flow * Update Flow * Update Flow * add image recaptioning * Fix bug in t2v * update train_reward_lora.py * update reward training * Update V5.1 and mix multi text_encoders to one pipeline * Update V5.1 training Code * Update ComfyUI * Update Comment * Delete files * update reward training * Update Readme * fix extract frames in compute_semantic_consistency * Update Readme && Remove to in prediction * Update Demo * Update Readme * Update ui * support vae gradient checkpointing in reward training * Update Training Readme --------- Co-authored-by: hkunzhe <huangkunzhe.hkz@alibaba-inc.com>
27 lines
326 B
Plaintext
27 lines
326 B
Plaintext
Pillow
|
|
einops
|
|
safetensors
|
|
timm
|
|
tomesd
|
|
torch>=2.1.2
|
|
torchdiffeq
|
|
torchsde
|
|
decord
|
|
datasets
|
|
numpy
|
|
scikit-image
|
|
opencv-python
|
|
omegaconf
|
|
SentencePiece
|
|
albumentations
|
|
imageio[ffmpeg]
|
|
imageio[pyav]
|
|
tensorboard
|
|
beautifulsoup4
|
|
ftfy
|
|
func_timeout
|
|
accelerate>=0.25.0
|
|
gradio>=3.41.2,<=3.48.0
|
|
diffusers>=0.30.1,<=0.31.0
|
|
transformers>=4.46.2
|