Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- **Without preservation/class:**
- export MODEL_NAME="CompVis/stable-diffusion-v1-4"
- export INSTANCE_DIR="training"
- export OUTPUT_DIR="my_model"
- accelerate launch train_dreambooth.py \
- --pretrained_model_name_or_path=$MODEL_NAME \
- --instance_data_dir=$INSTANCE_DIR \
- --output_dir=$OUTPUT_DIR \
- --instance_prompt="cialde" \
- --resolution=512 \
- --train_batch_size=1 \
- --learning_rate=5e-6 \
- --lr_scheduler="constant" \
- --lr_warmup_steps=0 \
- --gradient_accumulation_steps=2 --gradient_checkpointing \
- --use_8bit_adam \
- --max_train_steps=2000
- **With preservation and class:**
- export LD_LIBRARY_PATH=/usr/lib/wsl/lib:$LD_LIBRARY_PATH
- export MODEL_NAME="CompVis/stable-diffusion-v1-4"
- export INSTANCE_DIR="training"
- export CLASS_DIR="classes"
- export OUTPUT_DIR="my_model"
- accelerate launch train_dreambooth.py \
- --pretrained_model_name_or_path=$MODEL_NAME \
- --instance_data_dir=$INSTANCE_DIR \
- --class_data_dir=$CLASS_DIR \
- --output_dir=$OUTPUT_DIR \
- --with_prior_preservation --prior_loss_weight=1.0 \
- --instance_prompt="cialde" \
- --class_prompt="guy" \
- --resolution=512 \
- --train_batch_size=1 \
- --learning_rate=5e-6 \
- --lr_scheduler="constant" \
- --lr_warmup_steps=0 \
- --num_class_images=200 \
- --gradient_accumulation_steps=2 --gradient_checkpointing \
- --use_8bit_adam \
- --max_train_steps=2000
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement