Skip to content

GitLab

Commit b4437e88 authored by Maruyama_Aya's avatar Maruyama_Aya
Browse files

fixed port

parent 79c9f776
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/images/dreambooth/colossalai.sh
View file @ b4437e88
...@@ -2,7 +2,7 @@ HF_DATASETS_OFFLINE=1 ...@@ -2,7 +2,7 @@ HF_DATASETS_OFFLINE=1
TRANSFORMERS_OFFLINE=1 TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1 DIFFUSERS_OFFLINE=1
torchrun --nproc_per_node 4 --master_port=25641 train_dreambooth_colossalai.py \ torchrun --nproc_per_node 4 --standalone 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" \
......
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