This commit is contained in:
sayakpaul
2024-11-19 18:38:05 +05:30
parent 5208c59394
commit 9eea656fa0
2 changed files with 4 additions and 4 deletions
+1 -1
View File
@@ -26,7 +26,7 @@ cmd="accelerate launch --config_file deepspeed.yaml --gpu_ids $GPU_IDS text_to_v
--rank 64 \
--lora_alpha 64 \
--mixed_precision bf16 \
--output_dir mochi-lora \
--output_dir /raid/.cache/huggingface/sayak/mochi-lora/ \
--max_num_frames 84 \
--train_batch_size 1 \
--dataloader_num_workers 4 \