Skip to content

GitLab

Commit 79c9f776 authored by Maruyama_Aya's avatar Maruyama_Aya
Browse files

fixed port

parent d3379f0b
Hide whitespace changes
Inline Side-by-side
Showing with 1 addition and 1 deletion
examples/images/dreambooth/test_ci.sh
View file @ 79c9f776
...@@ -7,7 +7,7 @@ TRANSFORMERS_OFFLINE=1 ...@@ -7,7 +7,7 @@ TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1 DIFFUSERS_OFFLINE=1
for plugin in "torch_ddp" "torch_ddp_fp16" "gemini" "low_level_zero"; do for plugin in "torch_ddp" "torch_ddp_fp16" "gemini" "low_level_zero"; do
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="Your Pretrained Model Path" \ --pretrained_model_name_or_path="Your Pretrained Model Path" \
--instance_data_dir="Your Input Pics Path" \ --instance_data_dir="Your Input Pics Path" \
--output_dir="path-to-save-model" \ --output_dir="path-to-save-model" \
......
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