Sayak Paul 6c00cf094b fix: resuming from a checkpoint when using deepspeed. (#38) ...

* fix: resuming from a checkpoint when using deepspeed.

* remove changes to prepare_dataset.py

* propagate to others.

* tackle gradnorm.

---------

Co-authored-by: Aryan <aryan@huggingface.co>

2024-10-20 02:56:32 +05:30

..

text_encoder

CogVideoX LoRA and full finetuning (#1)

2024-10-01 08:05:10 +05:30

args.py

add VideoDatasetWithResizeAndRectangleCrop dataset resize crop (#13)

2024-10-10 01:55:12 +05:30

cogvideox_image_to_video_lora.py

fix: resuming from a checkpoint when using deepspeed. (#38)

2024-10-20 02:56:32 +05:30

cogvideox_text_to_video_lora.py

fix: resuming from a checkpoint when using deepspeed. (#38)

2024-10-20 02:56:32 +05:30

cogvideox_text_to_video_sft.py

fix: resuming from a checkpoint when using deepspeed. (#38)

2024-10-20 02:56:32 +05:30

dataset.py

more dataset fixes from stashed changes (#49)

2024-10-19 06:30:26 +05:30

prepare_dataset.py

more dataset fixes from stashed changes (#49)

2024-10-19 06:30:26 +05:30

utils.py

Low-bit memory optimizers, CpuOffloadOptimizer, Memory Reports (#3)

2024-10-04 18:18:54 +05:30