Describe the bug
Currently the Textual Inversion example uses Stable Diffusion v1.5 in the command, but the instructions say Stable Diffusion v1.4
|
You need to accept the model license before downloading or using the weights. In this example we'll use model version `v1-4`, so you'll need to visit [its card](https://huggingface.co/CompVis/stable-diffusion-v1-4), read the license and tick the checkbox if you agree. |
|
export MODEL_NAME="runwayml/stable-diffusion-v1-5" |
|
export DATA_DIR="path-to-dir-containing-images" |
|
|
|
accelerate launch textual_inversion.py \ |
|
--pretrained_model_name_or_path=$MODEL_NAME \ |
|
--train_data_dir=$DATA_DIR \ |
|
--learnable_property="object" \ |
|
--placeholder_token="<cat-toy>" --initializer_token="toy" \ |
|
--resolution=512 \ |
|
--train_batch_size=1 \ |
|
--gradient_accumulation_steps=4 \ |
|
--max_train_steps=3000 \ |
|
--learning_rate=5.0e-04 --scale_lr \ |
|
--lr_scheduler="constant" \ |
|
--lr_warmup_steps=0 \ |
|
--output_dir="textual_inversion_cat" |
The instruction should be updated and link to https://huggingface.co/runwayml/stable-diffusion-v1-5 instead.
Reproduction
No response
Logs
No response
System Info
NA
