storytold/FineTrainers-Conditioning
2
0
Fork 0
mirror of https://github.com/storytold/FineTrainers-Conditioning.git synced 2026-10-24 08:25:15 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply suggestions from code review

Browse Source 
Co-authored-by: Aryan <aryan@huggingface.co>
This commit is contained in:
Sayak Paul
2024-11-29 13:11:08 +05:30
committed by GitHub
parent 9c86706cbb
commit 38f157c6d0
2 changed files with 3 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/text_to_video_lora.py
+2 -2
View File
@@ -89,9 +89,9 @@ def save_model_card(
## Model description ## Model description
This is a lora finetune of the Moch-1 preview model `{base_model}`. This is a lora finetune of the Mochi-1 preview model `{base_model}`.
The model was trained using [CogVideoX Factory](https://github.com/a-r-r-o-w/cogvideox-factory) - a repository containing memory-optimized training scripts for the CogVideoX, Mochi family of models using [TorchAO](https://github.com/pytorch/ao) and [DeepSpeed](https://github.com/microsoft/DeepSpeed). The scripts were adopted from [CogVideoX Diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/cogvideo/train_cogvideox_lora.py). The model was trained using [CogVideoX Factory](https://github.com/a-r-r-o-w/cogvideox-factory) - a repository containing memory-optimized training scripts for the CogVideoX and Mochi family of models using [TorchAO](https://github.com/pytorch/ao) and [DeepSpeed](https://github.com/microsoft/DeepSpeed). The scripts were adopted from [CogVideoX Diffusers trainer](https://github.com/huggingface/diffusers/blob/main/examples/cogvideo/train_cogvideox_lora.py).
## Download model ## Download model
training/mochi-1/train.sh
+1 -1
View File
@@ -8,7 +8,7 @@ DATA_ROOT="videos_prepared"
MODEL="genmo/mochi-1-preview" MODEL="genmo/mochi-1-preview"
OUTPUT_PATH="mochi-lora" OUTPUT_PATH="mochi-lora"
cmd="CUDA_VISIBLE_DEVICES=$GPU_IDS python text_to_video_lora_simple.py \ cmd="CUDA_VISIBLE_DEVICES=$GPU_IDS python text_to_video_lora.py \
--pretrained_model_name_or_path $MODEL \ --pretrained_model_name_or_path $MODEL \
--cast_dit \ --cast_dit \
--data_root $DATA_ROOT \ --data_root $DATA_ROOT \