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

updates

Browse Source
This commit is contained in:
sayakpaul
2024-11-22 23:34:17 +05:30
parent 9a15eaee41
commit 2dbddd566e
5 changed files with 87 additions and 103 deletions
Download Patch File Download Diff File Expand all files Collapse all files
training/mochi-1/train.sh
+3 -7
View File
@@ -16,21 +16,17 @@ cmd="accelerate launch --config_file deepspeed.yaml --gpu_ids $GPU_IDS text_to_v
--id_token BW_STYLE \
--height_buckets 480 \
--width_buckets 848 \
--frame_buckets 84 \
--frame_buckets 85 \
--load_tensors \
--validation_prompt \"BW_STYLE A black and white animated scene unfolds with an anthropomorphic goat surrounded by musical notes and symbols, suggesting a playful environment. Mickey Mouse appears, leaning forward in curiosity as the goat remains still. The goat then engages with Mickey, who bends down to converse or react. The dynamics shift as Mickey grabs the goat, potentially in surprise or playfulness, amidst a minimalistic background. The scene captures the evolving relationship between the two characters in a whimsical, animated setting, emphasizing their interactions and emotions\" \
--validation_prompt_separator ::: \
--num_validation_videos 1 \
--validation_epochs 1 \
--seed 42 \
--rank 64 \
--lora_alpha 64 \
--mixed_precision bf16 \
--output_dir /raid/.cache/huggingface/sayak/mochi-lora/ \
--max_num_frames 84 \
--max_num_frames 85 \
--train_batch_size 1 \
--dataloader_num_workers 4 \
--max_train_steps 500 \
--max_train_steps 10 \
--checkpointing_steps 50 \
--gradient_accumulation_steps 4 \
--gradient_checkpointing \