├── .gitignore ├── .lightningignore ├── .pre-commit-config.yaml ├── LICENSE ├── README.md ├── benchmarks ├── README.md ├── locustfile.py ├── sd_python.py └── sd_triton.py ├── diffusion2 ├── app.py ├── app_interactive.py └── requirements.txt ├── lightning_diffusion ├── __about__.py ├── __init__.py ├── base_diffusion.py ├── client.py ├── datasets.py ├── diffusion_juspty.py ├── diffusion_serve.py ├── diffusion_serve_work.py ├── dreambooth.py ├── lambda_work.py ├── lite_finetuner.py ├── model_cloud │ ├── __init__.py │ ├── cloud_api.py │ ├── save.py │ └── utils.py ├── models.py ├── safety_checker.py └── setup_tools.py ├── requirements.txt ├── serve_diffusion_component.py ├── serve_dreambooth_component.py ├── setup.py └── tests ├── __init__.py ├── requirements.txt └── test_safety_filter.py /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/.gitignore -------------------------------------------------------------------------------- /.lightningignore: -------------------------------------------------------------------------------- 1 | 768-v-ema.ckpt 2 | .idea/ 3 | build/ 4 | -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/LICENSE -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/README.md -------------------------------------------------------------------------------- /benchmarks/README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/benchmarks/README.md -------------------------------------------------------------------------------- /benchmarks/locustfile.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/benchmarks/locustfile.py -------------------------------------------------------------------------------- /benchmarks/sd_python.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/benchmarks/sd_python.py -------------------------------------------------------------------------------- /benchmarks/sd_triton.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/benchmarks/sd_triton.py -------------------------------------------------------------------------------- /diffusion2/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/diffusion2/app.py -------------------------------------------------------------------------------- /diffusion2/app_interactive.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/diffusion2/app_interactive.py -------------------------------------------------------------------------------- /diffusion2/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/diffusion2/requirements.txt -------------------------------------------------------------------------------- /lightning_diffusion/__about__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/__about__.py -------------------------------------------------------------------------------- /lightning_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/__init__.py -------------------------------------------------------------------------------- /lightning_diffusion/base_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/base_diffusion.py -------------------------------------------------------------------------------- /lightning_diffusion/client.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/client.py -------------------------------------------------------------------------------- /lightning_diffusion/datasets.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/datasets.py -------------------------------------------------------------------------------- /lightning_diffusion/diffusion_juspty.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/diffusion_juspty.py -------------------------------------------------------------------------------- /lightning_diffusion/diffusion_serve.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/diffusion_serve.py -------------------------------------------------------------------------------- /lightning_diffusion/diffusion_serve_work.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/diffusion_serve_work.py -------------------------------------------------------------------------------- /lightning_diffusion/dreambooth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/dreambooth.py -------------------------------------------------------------------------------- /lightning_diffusion/lambda_work.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/lambda_work.py -------------------------------------------------------------------------------- /lightning_diffusion/lite_finetuner.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/lite_finetuner.py -------------------------------------------------------------------------------- /lightning_diffusion/model_cloud/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/model_cloud/__init__.py -------------------------------------------------------------------------------- /lightning_diffusion/model_cloud/cloud_api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/model_cloud/cloud_api.py -------------------------------------------------------------------------------- /lightning_diffusion/model_cloud/save.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/model_cloud/save.py -------------------------------------------------------------------------------- /lightning_diffusion/model_cloud/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/model_cloud/utils.py -------------------------------------------------------------------------------- /lightning_diffusion/models.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/models.py -------------------------------------------------------------------------------- /lightning_diffusion/safety_checker.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/safety_checker.py -------------------------------------------------------------------------------- /lightning_diffusion/setup_tools.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/lightning_diffusion/setup_tools.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/requirements.txt -------------------------------------------------------------------------------- /serve_diffusion_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/serve_diffusion_component.py -------------------------------------------------------------------------------- /serve_dreambooth_component.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/serve_dreambooth_component.py -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/setup.py -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/requirements.txt: -------------------------------------------------------------------------------- 1 | pytest 2 | -------------------------------------------------------------------------------- /tests/test_safety_filter.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/Lightning-Universe/lightning-diffusion_component/HEAD/tests/test_safety_filter.py --------------------------------------------------------------------------------