Skip to content

Instantly share code, notes, and snippets.

@garystafford

garystafford/autotrain_dreambooth_command.py

Last active November 23, 2023 04:43
Show Gist options
  • Star 0 You must be signed in to star a gist
  • Fork 0 You must be signed in to fork a gist
  • Save garystafford/f05c06a49939c2421b964aebfbe6b62d to your computer and use it in GitHub Desktop.
Save garystafford/f05c06a49939c2421b964aebfbe6b62d to your computer and use it in GitHub Desktop.
Download ZIP
Raw
autotrain_dreambooth_command.py
!autotrain dreambooth \
--model ${MODEL_NAME} \
--project-name ${PROJECT_NAME} \
--image-path "${IMAGE_PATH}" \
--prompt "${INSTANCE_PROMPT}" \
--class-prompt "${CLASS_PROMPT}" \
--resolution ${RESOLUTION} \
--batch-size ${BATCH_SIZE} \
--num-steps ${NUM_STEPS} \
--gradient-accumulation ${GRADIENT_ACCUMULATION} \
--lr ${LEARNING_RATE} \
--fp16 \
--gradient-checkpointing
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment