├── .github ├── PULL_REQUEST_TEMPLATE.md └── workflows │ ├── lint.yml │ └── pypi-publish.yml ├── .gitignore ├── .pre-commit-config.yaml ├── .swift-format ├── CONTRIBUTING.md ├── LICENSE ├── Makefile ├── Package.resolved ├── Package.swift ├── README.md ├── assets └── diffusionkit.png ├── python ├── __init__.py └── src │ ├── __init__.py │ └── diffusionkit │ ├── __init__.py │ ├── mlx │ ├── __init__.py │ ├── clip.py │ ├── config.py │ ├── mmdit.py │ ├── model_io.py │ ├── sampler.py │ ├── scripts │ │ ├── __init__.py │ │ └── generate_images.py │ ├── t5.py │ ├── tokenizer.py │ └── vae.py │ ├── tests │ ├── __init__.py │ ├── mlx │ │ ├── __init__.py │ │ └── test_diffusion_pipeline.py │ └── torch2coreml │ │ ├── __init__.py │ │ ├── test_mmdit.py │ │ └── test_vae.py │ ├── torch │ ├── __init__.py │ ├── mmdit.py │ ├── model_io.py │ └── vae.py │ └── utils.py ├── requirements.txt ├── setup.py └── swift ├── .gitignore ├── Sources ├── DiffusionKit │ └── DiffusionKit.swift └── DiffusionKitMLX │ └── DiffusionKitMLX.swift └── Tests ├── DiffusionKitMLXTests └── DiffusionKitMLXTests.swift └── DiffusionKitTests └── DiffusionKitTests.swift /.github/PULL_REQUEST_TEMPLATE.md: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /.github/workflows/lint.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/.github/workflows/lint.yml -------------------------------------------------------------------------------- /.github/workflows/pypi-publish.yml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/.github/workflows/pypi-publish.yml -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/.gitignore -------------------------------------------------------------------------------- /.pre-commit-config.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/.pre-commit-config.yaml -------------------------------------------------------------------------------- /.swift-format: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/.swift-format -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/LICENSE -------------------------------------------------------------------------------- /Makefile: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/Makefile -------------------------------------------------------------------------------- /Package.resolved: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/Package.resolved -------------------------------------------------------------------------------- /Package.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/Package.swift -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/README.md -------------------------------------------------------------------------------- /assets/diffusionkit.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/assets/diffusionkit.png -------------------------------------------------------------------------------- /python/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/diffusionkit/__init__.py: -------------------------------------------------------------------------------- 1 | import os 2 | 3 | os.environ["TOKENIZERS_PARALLELISM"] = "false" 4 | -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/__init__.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/clip.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/clip.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/config.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/config.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/mmdit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/mmdit.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/model_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/model_io.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/sampler.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/scripts/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/scripts/generate_images.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/scripts/generate_images.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/t5.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/t5.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/tokenizer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/tokenizer.py -------------------------------------------------------------------------------- /python/src/diffusionkit/mlx/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/mlx/vae.py -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/mlx/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/mlx/test_diffusion_pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/tests/mlx/test_diffusion_pipeline.py -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/torch2coreml/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/tests/torch2coreml/__init__.py -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/torch2coreml/test_mmdit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/tests/torch2coreml/test_mmdit.py -------------------------------------------------------------------------------- /python/src/diffusionkit/tests/torch2coreml/test_vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/tests/torch2coreml/test_vae.py -------------------------------------------------------------------------------- /python/src/diffusionkit/torch/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /python/src/diffusionkit/torch/mmdit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/torch/mmdit.py -------------------------------------------------------------------------------- /python/src/diffusionkit/torch/model_io.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/torch/model_io.py -------------------------------------------------------------------------------- /python/src/diffusionkit/torch/vae.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/torch/vae.py -------------------------------------------------------------------------------- /python/src/diffusionkit/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/python/src/diffusionkit/utils.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/setup.py -------------------------------------------------------------------------------- /swift/.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/swift/.gitignore -------------------------------------------------------------------------------- /swift/Sources/DiffusionKit/DiffusionKit.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/swift/Sources/DiffusionKit/DiffusionKit.swift -------------------------------------------------------------------------------- /swift/Sources/DiffusionKitMLX/DiffusionKitMLX.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/swift/Sources/DiffusionKitMLX/DiffusionKitMLX.swift -------------------------------------------------------------------------------- /swift/Tests/DiffusionKitMLXTests/DiffusionKitMLXTests.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/swift/Tests/DiffusionKitMLXTests/DiffusionKitMLXTests.swift -------------------------------------------------------------------------------- /swift/Tests/DiffusionKitTests/DiffusionKitTests.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/argmaxinc/DiffusionKit/HEAD/swift/Tests/DiffusionKitTests/DiffusionKitTests.swift --------------------------------------------------------------------------------