Skip to content

GitLab

Commit b56c7f42 authored by Maruyama_Aya's avatar Maruyama_Aya
Browse files

update shell file

parent 176010f2
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/tutorial/new_api/dreambooth/colossalai.sh
View file @ b56c7f42
...@@ -6,7 +6,7 @@ torchrun --nproc_per_node 4 --master_port=25641 train_dreambooth_colossalai.py \ ...@@ -6,7 +6,7 @@ torchrun --nproc_per_node 4 --master_port=25641 train_dreambooth_colossalai.py \
--pretrained_model_name_or_path="Path_to_your_model" \ --pretrained_model_name_or_path="Path_to_your_model" \
--instance_data_dir="Path_to_your_training_image" \ --instance_data_dir="Path_to_your_training_image" \
--output_dir="Path_to_your_save_dir" \ --output_dir="Path_to_your_save_dir" \
--instance_prompt="keli" \ --instance_prompt="your prompt" \
--resolution=512 \ --resolution=512 \
--plugin="gemini" \ --plugin="gemini" \
--train_batch_size=1 \ --train_batch_size=1 \
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment