Skip to content

GitLab

Unverified Commit 30e7c78a authored by Suraj Patil's avatar Suraj Patil Committed by GitHub
Browse files

Update README.md

parent d0d3e24e
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
examples/textual_inversion/README.md
View file @ 30e7c78a
...@@ -49,7 +49,7 @@ accelerate launch textual_inversion.py \ ...@@ -49,7 +49,7 @@ accelerate launch textual_inversion.py \
--placeholder_token="<cat-toy>" --initializer_token="toy" \ --placeholder_token="<cat-toy>" --initializer_token="toy" \
--resolution=512 \ --resolution=512 \
--train_batch_size=1 \ --train_batch_size=1 \
--gradient_accumulation_steps=2 \ --gradient_accumulation_steps=4 \
--max_train_steps=3000 \ --max_train_steps=3000 \
--learning_rate=5.0e-04 --scale_lr \ --learning_rate=5.0e-04 --scale_lr \
--lr_scheduler="constant" \ --lr_scheduler="constant" \
...@@ -79,4 +79,4 @@ with autocast("cuda"): ...@@ -79,4 +79,4 @@ with autocast("cuda"):
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)["sample"][0] image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)["sample"][0]
image.save("cat-backpack.png") image.save("cat-backpack.png")
``` ```
\ No newline at end of file
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