├── .github └── pull_request_template.md ├── .gitignore ├── ACKNOWLEDGEMENTS ├── CODE_OF_CONDUCT.md ├── CONTRIBUTING.md ├── LICENSE.md ├── Package.swift ├── README.md ├── assets ├── a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space │ ├── randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png │ ├── randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png │ ├── randomSeed_123456789_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_123456789_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_123456789_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_93_computeUnit_ALL_modelVersion_stabilityai_stable-diffusion-2-base.png │ ├── randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png │ └── randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png ├── a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space │ ├── randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png │ ├── randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png │ ├── randomSeed_13_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_13_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_13_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png │ ├── randomSeed_93_computeUnit_ALL_modelVersion_runwayml_stable-diffusion-v1-5.png │ ├── randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png │ └── randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png ├── controlnet_readme_reel.png ├── float16_cpuandne_readmereel.png ├── float16_gpu_readmereel.png ├── mbp │ ├── a_high_quality_photo_of_a_surfing_dog.7667.final_3.41-bits.png │ ├── a_high_quality_photo_of_a_surfing_dog.7667.final_4.50-bits.png │ ├── a_high_quality_photo_of_a_surfing_dog.7667.final_6.55-bits.png │ ├── a_high_quality_photo_of_a_surfing_dog.7667.final_float16_original.png │ ├── runwayml_stable-diffusion-v1-5_psnr_vs_size.png │ ├── stabilityai_stable-diffusion-2-1-base_psnr_vs_size.png │ └── stabilityai_stable-diffusion-xl-base-1.0_psnr_vs_size.png ├── palette6_cpuandne_readmereel.png └── readme_reel.png ├── python_coreml_stable_diffusion ├── __init__.py ├── _version.py ├── activation_quantization.py ├── attention.py ├── chunk_mlprogram.py ├── controlnet.py ├── coreml_model.py ├── layer_norm.py ├── mixed_bit_compression_apply.py ├── mixed_bit_compression_pre_analysis.py ├── multilingual_projection.py ├── pipeline.py ├── torch2coreml.py └── unet.py ├── requirements.txt ├── setup.py ├── swift ├── StableDiffusion │ ├── pipeline │ │ ├── CGImage+vImage.swift │ │ ├── ControlNet.swift │ │ ├── DPMSolverMultistepScheduler.swift │ │ ├── Decoder.swift │ │ ├── DiscreteFlowScheduler.swift │ │ ├── Encoder.swift │ │ ├── ManagedMLModel.swift │ │ ├── MultiModalDiffusionTransformer.swift │ │ ├── MultilingualTextEncoder.swift │ │ ├── NumPyRandomSource.swift │ │ ├── NvRandomSource.swift │ │ ├── RandomSource.swift │ │ ├── ResourceManaging.swift │ │ ├── SafetyChecker.swift │ │ ├── SampleTimer.swift │ │ ├── Scheduler.swift │ │ ├── StableDiffusion3Pipeline+Resources.swift │ │ ├── StableDiffusion3Pipeline.swift │ │ ├── StableDiffusionPipeline+Resources.swift │ │ ├── StableDiffusionPipeline.Configuration.swift │ │ ├── StableDiffusionPipeline.swift │ │ ├── StableDiffusionXL+Resources.swift │ │ ├── StableDiffusionXLPipeline.swift │ │ ├── TextEncoder.swift │ │ ├── TextEncoderT5.swift │ │ ├── TextEncoderXL.swift │ │ ├── TorchRandomSource.swift │ │ └── Unet.swift │ └── tokenizer │ │ ├── BPETokenizer+Reading.swift │ │ ├── BPETokenizer.swift │ │ └── T5Tokenizer.swift ├── StableDiffusionCLI │ └── main.swift └── StableDiffusionTests │ ├── Resources │ ├── merges.txt │ └── vocab.json │ └── StableDiffusionTests.swift └── tests ├── __init__.py └── test_stable_diffusion.py /.github/pull_request_template.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/.github/pull_request_template.md -------------------------------------------------------------------------------- /.gitignore: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/.gitignore -------------------------------------------------------------------------------- /ACKNOWLEDGEMENTS: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/ACKNOWLEDGEMENTS -------------------------------------------------------------------------------- /CODE_OF_CONDUCT.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/CODE_OF_CONDUCT.md -------------------------------------------------------------------------------- /CONTRIBUTING.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/CONTRIBUTING.md -------------------------------------------------------------------------------- /LICENSE.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/LICENSE.md -------------------------------------------------------------------------------- /Package.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/Package.swift -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/README.md -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_123456789_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_ALL_modelVersion_stabilityai_stable-diffusion-2-base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_ALL_modelVersion_stabilityai_stable-diffusion-2-base.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_dragon_in_space/randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_11_computeUnit_CPU_AND_GPU_modelVersion_stabilityai_stable-diffusion-2-base.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_11_computeUnit_CPU_AND_NE_modelVersion_stabilityai_stable-diffusion-2-base.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_ALL_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_CPU_AND_GPU_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_13_computeUnit_CPU_AND_NE_modelVersion_CompVis_stable-diffusion-v1-4.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_ALL_modelVersion_runwayml_stable-diffusion-v1-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_ALL_modelVersion_runwayml_stable-diffusion-v1-5.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_CPU_AND_GPU_modelVersion_runwayml_stable-diffusion-v1-5.png -------------------------------------------------------------------------------- /assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/a_high_quality_photo_of_an_astronaut_riding_a_horse_in_space/randomSeed_93_computeUnit_CPU_AND_NE_modelVersion_runwayml_stable-diffusion-v1-5.png -------------------------------------------------------------------------------- /assets/controlnet_readme_reel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/controlnet_readme_reel.png -------------------------------------------------------------------------------- /assets/float16_cpuandne_readmereel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/float16_cpuandne_readmereel.png -------------------------------------------------------------------------------- /assets/float16_gpu_readmereel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/float16_gpu_readmereel.png -------------------------------------------------------------------------------- /assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_3.41-bits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_3.41-bits.png -------------------------------------------------------------------------------- /assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_4.50-bits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_4.50-bits.png -------------------------------------------------------------------------------- /assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_6.55-bits.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_6.55-bits.png -------------------------------------------------------------------------------- /assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_float16_original.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/a_high_quality_photo_of_a_surfing_dog.7667.final_float16_original.png -------------------------------------------------------------------------------- /assets/mbp/runwayml_stable-diffusion-v1-5_psnr_vs_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/runwayml_stable-diffusion-v1-5_psnr_vs_size.png -------------------------------------------------------------------------------- /assets/mbp/stabilityai_stable-diffusion-2-1-base_psnr_vs_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/stabilityai_stable-diffusion-2-1-base_psnr_vs_size.png -------------------------------------------------------------------------------- /assets/mbp/stabilityai_stable-diffusion-xl-base-1.0_psnr_vs_size.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/mbp/stabilityai_stable-diffusion-xl-base-1.0_psnr_vs_size.png -------------------------------------------------------------------------------- /assets/palette6_cpuandne_readmereel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/palette6_cpuandne_readmereel.png -------------------------------------------------------------------------------- /assets/readme_reel.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/assets/readme_reel.png -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/__init__.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/_version.py: -------------------------------------------------------------------------------- 1 | __version__ = "1.1.0" 2 | -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/activation_quantization.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/activation_quantization.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/attention.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/chunk_mlprogram.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/chunk_mlprogram.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/controlnet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/controlnet.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/coreml_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/coreml_model.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/layer_norm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/layer_norm.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/mixed_bit_compression_apply.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/mixed_bit_compression_apply.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/mixed_bit_compression_pre_analysis.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/multilingual_projection.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/multilingual_projection.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/pipeline.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/pipeline.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/torch2coreml.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/torch2coreml.py -------------------------------------------------------------------------------- /python_coreml_stable_diffusion/unet.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/python_coreml_stable_diffusion/unet.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/setup.py -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/CGImage+vImage.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/CGImage+vImage.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/ControlNet.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/ControlNet.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/DPMSolverMultistepScheduler.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/Decoder.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/Decoder.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/DiscreteFlowScheduler.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/DiscreteFlowScheduler.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/Encoder.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/Encoder.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/ManagedMLModel.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/ManagedMLModel.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/MultiModalDiffusionTransformer.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/MultiModalDiffusionTransformer.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/MultilingualTextEncoder.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/MultilingualTextEncoder.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/NumPyRandomSource.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/NumPyRandomSource.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/NvRandomSource.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/NvRandomSource.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/RandomSource.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/RandomSource.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/ResourceManaging.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/ResourceManaging.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/SafetyChecker.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/SafetyChecker.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/SampleTimer.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/SampleTimer.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/Scheduler.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/Scheduler.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusion3Pipeline+Resources.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusion3Pipeline+Resources.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusion3Pipeline.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusion3Pipeline.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusionPipeline+Resources.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusionPipeline+Resources.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusionPipeline.Configuration.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusionPipeline.Configuration.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusionPipeline.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusionPipeline.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusionXL+Resources.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusionXL+Resources.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/StableDiffusionXLPipeline.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/StableDiffusionXLPipeline.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/TextEncoder.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/TextEncoder.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/TextEncoderT5.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/TextEncoderT5.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/TextEncoderXL.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/TextEncoderXL.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/TorchRandomSource.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/TorchRandomSource.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/pipeline/Unet.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/pipeline/Unet.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/tokenizer/BPETokenizer+Reading.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/tokenizer/BPETokenizer+Reading.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/tokenizer/BPETokenizer.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/tokenizer/BPETokenizer.swift -------------------------------------------------------------------------------- /swift/StableDiffusion/tokenizer/T5Tokenizer.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusion/tokenizer/T5Tokenizer.swift -------------------------------------------------------------------------------- /swift/StableDiffusionCLI/main.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusionCLI/main.swift -------------------------------------------------------------------------------- /swift/StableDiffusionTests/Resources/merges.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusionTests/Resources/merges.txt -------------------------------------------------------------------------------- /swift/StableDiffusionTests/Resources/vocab.json: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusionTests/Resources/vocab.json -------------------------------------------------------------------------------- /swift/StableDiffusionTests/StableDiffusionTests.swift: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/swift/StableDiffusionTests/StableDiffusionTests.swift -------------------------------------------------------------------------------- /tests/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /tests/test_stable_diffusion.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/apple/ml-stable-diffusion/HEAD/tests/test_stable_diffusion.py --------------------------------------------------------------------------------