Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
OpenDAS
ColossalAI
Commits
4fc8bc68
Commit
4fc8bc68
authored
Jun 07, 2023
by
Maruyama_Aya
Browse files
modify file path
parent
b4437e88
Changes
3
Hide whitespace changes
Inline
Side-by-side
Showing
3 changed files
with
11 additions
and
11 deletions
+11
-11
examples/images/dreambooth/colossalai.sh
examples/images/dreambooth/colossalai.sh
+4
-4
examples/images/dreambooth/dreambooth.sh
examples/images/dreambooth/dreambooth.sh
+3
-3
examples/images/dreambooth/test_ci.sh
examples/images/dreambooth/test_ci.sh
+4
-4
No files found.
examples/images/dreambooth/colossalai.sh
View file @
4fc8bc68
...
@@ -3,10 +3,10 @@ TRANSFORMERS_OFFLINE=1
...
@@ -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
\
...
...
examples/images/dreambooth/dreambooth.sh
View file @
4fc8bc68
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
\
...
...
examples/images/dreambooth/test_ci.sh
View file @
4fc8bc68
...
@@ -8,10 +8,10 @@ DIFFUSERS_OFFLINE=1
...
@@ -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
\
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment