1
0
mirror of https://github.com/huggingface/diffusers.git synced 2026-01-29 07:22:12 +03:00

Update README.md

This commit is contained in:
Suraj Patil
2022-09-02 14:29:27 +05:30
committed by GitHub
parent d0d3e24ec1
commit 30e7c78ac3

View File

@@ -49,7 +49,7 @@ accelerate launch textual_inversion.py \
--placeholder_token="<cat-toy>" --initializer_token="toy" \
--resolution=512 \
--train_batch_size=1 \
--gradient_accumulation_steps=2 \
--gradient_accumulation_steps=4 \
--max_train_steps=3000 \
--learning_rate=5.0e-04 --scale_lr \
--lr_scheduler="constant" \
@@ -79,4 +79,4 @@ with autocast("cuda"):
image = pipe(prompt, num_inference_steps=50, guidance_scale=7.5)["sample"][0]
image.save("cat-backpack.png")
```
```