Advertisement
Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- ***My path folder (WSL2 + Ubuntu):***
- https://i.imgur.com/UwzYmxK.jpg
- ***My train config file:***
- 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="AlbertoFinstance" \
- --class_prompt="man" \
- --resolution=512 \
- --train_batch_size=1 \
- --learning_rate=5e-6 \
- --lr_scheduler="constant" \
- --lr_warmup_steps=0 \
- --gradient_accumulation_steps=1 --gradient_checkpointing \
- --use_8bit_adam \
- --max_train_steps=2500
Advertisement
Add Comment
Please, Sign In to add comment
Advertisement