├── .gitignore ├── LICENSE ├── README.md ├── notebooks ├── generate_class_priors.ipynb └── inference_dreambooth.ipynb ├── requirements.txt ├── scripts └── generate_experimental_images.py ├── src ├── __init__.py ├── constants.py ├── datasets.py ├── dreambooth_trainer.py └── utils.py └── train_dreambooth.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/.gitignore -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/README.md -------------------------------------------------------------------------------- /notebooks/generate_class_priors.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/notebooks/generate_class_priors.ipynb -------------------------------------------------------------------------------- /notebooks/inference_dreambooth.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/notebooks/inference_dreambooth.ipynb -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/requirements.txt -------------------------------------------------------------------------------- /scripts/generate_experimental_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/scripts/generate_experimental_images.py -------------------------------------------------------------------------------- /src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/constants.py: -------------------------------------------------------------------------------- 1 | PADDING_TOKEN = 49407 2 | MAX_PROMPT_LENGTH = 77 3 | -------------------------------------------------------------------------------- /src/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/src/datasets.py -------------------------------------------------------------------------------- /src/dreambooth_trainer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/src/dreambooth_trainer.py -------------------------------------------------------------------------------- /src/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/src/utils.py -------------------------------------------------------------------------------- /train_dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/sayakpaul/dreambooth-keras/HEAD/train_dreambooth.py --------------------------------------------------------------------------------