github
/
ColossalAI
mirror of https://github.com/hpcaitech/ColossalAI
1
0
Fork 0
Code Issues Projects Releases Wiki Activity

modify file path

pull/3905/head
Maruyama_Aya 2023-06-07 11:02:19 +08:00
parent b4437e88c3
commit 4fc8bc68ac
3 changed files with 11 additions and 11 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
examples/images/dreambooth/colossalai.sh
View File

@ -3,10 +3,10 @@ TRANSFORMERS_OFFLINE=1
DIFFUSERS_OFFLINE=1 DIFFUSERS_OFFLINE=1
torchrun --nproc_per_node 4 --standalone 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="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir="Path_to_your_training_image" \ --instance_data_dir="/data/dreambooth/Teyvat/data" \
--output_dir="Path_to_your_save_dir" \ --output_dir="./weight_output" \
--instance_prompt="your prompt" \ --instance_prompt="a picture of a dog" \
--resolution=512 \ --resolution=512 \
--plugin="gemini" \ --plugin="gemini" \
--train_batch_size=1 \ --train_batch_size=1 \

6
examples/images/dreambooth/dreambooth.sh
View File

@ -1,7 +1,7 @@
python train_dreambooth.py \ python train_dreambooth.py \
--pretrained_model_name_or_path= ## Your Model Path \ --pretrained_model_name_or_path="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir= ## Your Training Input Pics Path \ --instance_data_dir="/data/dreambooth/Teyvat/data" \
--output_dir="path-to-save-model" \ --output_dir="./weight_output" \
--instance_prompt="a photo of a dog" \ --instance_prompt="a photo of a dog" \
--resolution=512 \ --resolution=512 \
--train_batch_size=1 \ --train_batch_size=1 \

8
examples/images/dreambooth/test_ci.sh
View File

@ -8,10 +8,10 @@ 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 --standalone 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="/data/dreambooth/diffuser/stable-diffusion-v1-4" \
--instance_data_dir="Your Input Pics Path" \ --instance_data_dir="/data/dreambooth/Teyvat/data" \
--output_dir="path-to-save-model" \ --output_dir="./weight_output" \
--instance_prompt="your prompt" \ --instance_prompt="a picture of a dog" \
--resolution=512 \ --resolution=512 \
--plugin=$plugin \ --plugin=$plugin \
--train_batch_size=1 \ --train_batch_size=1 \