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
renzhc
diffusers_dcu
Commits
7d96b38b
Unverified
Commit
7d96b38b
authored
Jan 31, 2023
by
Sayak Paul
Committed by
GitHub
Jan 31, 2023
Browse files
[examples] Fix CLI argument in the launch script command for text2image with LoRA (#2171)
* Update README.md * Update README.md
parent
cedafb86
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
2 additions
and
2 deletions
+2
-2
examples/text_to_image/README.md
examples/text_to_image/README.md
+2
-2
No files found.
examples/text_to_image/README.md
View file @
7d96b38b
...
...
@@ -148,7 +148,7 @@ huggingface-cli login
Now we can start training!
```
bash
accelerate
--mixed_precision
=
"fp16"
launch
train_text_to_image_lora.py
\
accelerate
launch
--mixed_precision
=
"fp16"
train_text_to_image_lora.py
\
--pretrained_model_name_or_path
=
$MODEL_NAME
\
--dataset_name
=
$DATASET_NAME
--caption_column
=
"text"
\
--resolution
=
512
--random_flip
\
...
...
@@ -157,7 +157,7 @@ accelerate --mixed_precision="fp16" launch train_text_to_image_lora.py \
--learning_rate
=
1e-04
--lr_scheduler
=
"constant"
--lr_warmup_steps
=
0
\
--seed
=
42
\
--output_dir
=
"sd-pokemon-model-lora"
\
--
save_sample
_prompt
=
"cute dragon creature"
--report_to
=
"wandb"
--
validation
_prompt
=
"cute dragon creature"
--report_to
=
"wandb"
```
The above command will also run inference as fine-tuning progresses and log the results to Weights and Biases.
...
...
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