File size: 1,791 Bytes
e225ed6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 |
export MODEL_NAME="./checkpoint/van_gogh_erasure"
export DATA_DIR="./data/van_gogh_ti/train"
accelerate launch textual_inversion.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--train_data_dir=$DATA_DIR \
--learnable_property="style" \
--placeholder_token="<art-style>" \
--initializer_token="art" \
--resolution=512 \
--train_batch_size=1 \
--gradient_accumulation_steps=4 \
--max_train_steps=5000 \
--learning_rate=5.0e-04 \
--scale_lr \
--lr_scheduler="constant" \
--lr_warmup_steps=0 \
--push_to_hub \
--output_dir="./checkpoint/van_gogh_inversion"
export MODEL_NAME="./checkpoint/kilian_eng_erasure"
export DATA_DIR="./data/kilian_eng_ti/train"
accelerate launch textual_inversion.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--train_data_dir=$DATA_DIR \
--learnable_property="style" \
--placeholder_token="<art-style>" \
--initializer_token="art" \
--resolution=512 \
--train_batch_size=1 \
--gradient_accumulation_steps=4 \
--max_train_steps=5000 \
--learning_rate=5.0e-04 \
--scale_lr \
--lr_scheduler="constant" \
--lr_warmup_steps=0 \
--push_to_hub \
--output_dir="./checkpoint/kilian_eng_inversion"
export MODEL_NAME="./checkpoint/thomas_kinkade_erasure"
export DATA_DIR="./data/thomas_kinkade_ti/train"
accelerate launch textual_inversion.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--train_data_dir=$DATA_DIR \
--learnable_property="style" \
--placeholder_token="<art-style>" \
--initializer_token="art" \
--resolution=512 \
--train_batch_size=1 \
--gradient_accumulation_steps=4 \
--max_train_steps=5000 \
--learning_rate=5.0e-04 \
--scale_lr \
--lr_scheduler="constant" \
--lr_warmup_steps=0 \
--push_to_hub \
--output_dir="./checkpoint/thomas_kinkade_inversion"
|