Commit Graph
9 Commits
Author SHA1 Message Date
Bubbliiiingandhkunzhe 9f34f7c8c5 Update to V5.1 (#179)
* 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>
2025-01-22 15:30:47 +08:00
Bubbliiiing 45c298af71 Update Readme && Update lora cache && Lora merge with CUDA (#173)
* Update Readme

* Fix bug in training

* Update Readme

* Update Readme && Update lora cache

* Update Readme

* Update error info in sampler

* Fix Memory bug in 12b
2024-12-26 15:41:21 +08:00
Bubbliiiing 62de94e2f1 Update V5 (#128)
Update V5
2024-11-08 19:46:50 +08:00
Bubbliiiing d3b8bbbd14 Fix bug in v3 training (#98) 2024-08-22 16:03:14 +08:00
e4f1a4fe97 Update V4 version (#92)
* update train_lora && update deepspeed && update training with max token length

* fix bug in train.py

* fix bug in training_with_video_token_length

* update v2v && update v2v api

* add rope2d embedding precomputation; move text encoder to dataloader to reduce gpu memory consumpution

* add cuda multi-stream to speedup vae encode

* update new vae && new comfyui

* fix some bug in training code

* Add lcm lora (#89)

Co-authored-by: xuanyuan.lb <xuanyuan.lb@alibaba-inc.com>

* Update Training Code and fix bug in low vram mode

* fix bug in low vram mode

* update report

* update cfg

* actual text clip

---------

Co-authored-by: mengli.cml <mengli.cml@alibaba-inc.com>
Co-authored-by: liubo0902 <38622806+liubo0902@users.noreply.github.com>
Co-authored-by: xuanyuan.lb <xuanyuan.lb@alibaba-inc.com>
2024-08-19 11:22:17 +08:00
bubbliiiing 039d67acf1 support float16 && add reference && fix bug in training 2024-07-24 13:19:03 +08:00
bubbliiiing 2eef3f9f78 update v4 2024-07-18 16:33:56 +08:00
Bubbliiiing 616a35425d Fix the issue where Lora training cannot load state, fix the issue where training cannot eval (#66)
* rename the comfyui files and new readme

* fix bug in eval and load lora state dict
2024-07-18 15:47:51 +08:00
Bubbliiiing 6ed8619ad4 Rename the files (#45)
* add update readme

* update demo show and prompt

* rename the train code
2024-07-10 17:46:02 +08:00