Spaces:
Runtime error
Runtime error
File size: 601 Bytes
3e88ee7 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 |
export MODEL_NAME="CompVis/stable-diffusion-v1-4"
export OUTPUT_DIR="../../../models/imagic"
export INPUT_IMAGE="imgs/Official_portrait_of_Barack_Obama.jpg"
accelerate launch train_imagic.py \
--pretrained_model_name_or_path=$MODEL_NAME \
--output_dir=$OUTPUT_DIR \
--input_image=$INPUT_IMAGE \
--target_text="A photo of Barack Obama smiling with a big grin." \
--seed=3434554 \
--resolution=512 \
--mixed_precision="fp16" \
--use_8bit_adam \
--gradient_accumulation_steps=1 \
--emb_learning_rate=1e-3 \
--learning_rate=1e-6 \
--emb_train_steps=500 \
--max_train_steps=1000
|