Commit Graph

28 Commits

Author SHA1 Message Date
sayakpaul 84c17561eb fixes 2024-12-27 15:02:21 +05:30
sayakpaul 4461cdd98f tracker logging 2024-12-27 12:25:11 +05:30
Aryan a19401e84f update README; remove print; make style 2024-12-27 07:19:16 +01:00
sayakpaul 87af1a0113 Merge branch 'main' into support-deepspeed 2024-12-26 08:17:48 +05:30
sayakpaul 1d5e0566fb private 2024-12-26 08:01:16 +05:30
Sayak Paul 0309f3bf67 Apply suggestions from code review
Co-authored-by: Aryan <aryan@huggingface.co>
2024-12-26 08:00:04 +05:30
sayakpaul 0c0a0a5e4a scheduler 2024-12-26 07:55:07 +05:30
sayakpaul 28ecfe6b65 torch.is_tensor check 2024-12-26 07:49:51 +05:30
Aryan 2a594ea2db fix validation bug (#149) 2024-12-26 02:54:08 +05:30
Aryan 5757b1b886 fix grad norm related logging 2024-12-25 19:25:22 +01:00
sayakpaul 6b54867c3c fixes 2024-12-25 21:34:56 +05:30
Aryan 47e4aaac2f track grad norm for deepspeed 2024-12-25 15:04:16 +01:00
Aryan e2178deea0 fix deadlock; make style 2024-12-25 14:16:31 +01:00
sayakpaul d7aa50917f fixes 2024-12-25 12:07:50 +05:30
sayakpaul 69b585a992 fixes 2024-12-25 11:35:09 +05:30
sayakpaul 875cb72f63 address review feedback. 2024-12-24 21:20:37 +05:30
sayakpaul 6329e9ed07 support DeepSpeed. 2024-12-24 10:59:21 +05:30
Aryan 2858346b27 address review 2024-12-23 15:11:03 +01:00
Aryan 78712f60a5 update comment 2024-12-23 14:01:39 +01:00
Aryan 5261bed284 remove pipeline hooks 2024-12-23 13:58:37 +01:00
Aryan 86e1d1888a other condtions -> text conditions 2024-12-23 13:48:58 +01:00
Aryan 363841ae3f address some reviews; change hunyuan checkpoint repo-id 2024-12-23 12:54:52 +01:00
Aryan b045037850 make style 2024-12-21 15:19:03 +01:00
Aryan 0db7bf28aa update 2024-12-21 15:18:46 +01:00
Aryan 2da8cb5798 update 2024-12-21 12:24:33 +01:00
Aryan 66189c804b update 2024-12-20 10:03:39 +01:00
Aryan 223add1a59 Hunyuan Video LoRA (#126)
* add hunyuan-video lora support

* minor fixes; make style

* update readme

* update

* update

* Update README.md

* Update README.md

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>

* update

* update

* change move train scripts to internal directory

* update

---------

Co-authored-by: Sayak Paul <spsayakpaul@gmail.com>
2024-12-20 08:02:08 +05:30
Aryan 9ef58e2f3a LTX Video (#123)
* rename files

* ltx finetuning

* update

* update

* improvements

* make style

* gradient clipping

* update

* fix distributed inference

* update

* update
2024-12-19 04:27:36 +05:30