├── LICENSE ├── LICENSE-MODEL ├── MANIFEST.in ├── README.md ├── app-examples ├── app.py └── sd-compare │ ├── requirements.txt │ └── sd-compare.py ├── cats in hats.png ├── demo.ipynb ├── example.py ├── requirements.txt ├── setup.py ├── src ├── ldm1 │ ├── __init__.py │ ├── lr_scheduler.py │ ├── models │ │ ├── __init__.py │ │ ├── autoencoder.py │ │ └── diffusion │ │ │ ├── __init__.py │ │ │ ├── classifier.py │ │ │ ├── ddim.py │ │ │ ├── ddpm.py │ │ │ └── plms.py │ ├── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── diffusionmodules │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ ├── openaimodel.py │ │ │ └── util.py │ │ ├── distributions │ │ │ ├── __init__.py │ │ │ └── distributions.py │ │ ├── ema.py │ │ ├── encoders │ │ │ ├── __init__.py │ │ │ └── modules.py │ │ ├── image_degradation │ │ │ ├── __init__.py │ │ │ ├── bsrgan.py │ │ │ ├── bsrgan_light.py │ │ │ ├── utils │ │ │ │ └── test.png │ │ │ └── utils_image.py │ │ ├── losses │ │ │ ├── __init__.py │ │ │ ├── contperceptual.py │ │ │ └── vqperceptual.py │ │ └── x_transformer.py │ └── util.py ├── ldm2 │ ├── __init__.py │ ├── models │ │ ├── __init__.py │ │ ├── autoencoder.py │ │ └── diffusion │ │ │ ├── __init__.py │ │ │ ├── ddim.py │ │ │ ├── ddpm.py │ │ │ ├── dpm_solver │ │ │ ├── __init__.py │ │ │ ├── dpm_solver.py │ │ │ └── sampler.py │ │ │ ├── plms.py │ │ │ └── sampling_util.py │ ├── modules │ │ ├── __init__.py │ │ ├── attention.py │ │ ├── diffusionmodules │ │ │ ├── __init__.py │ │ │ ├── model.py │ │ │ ├── openaimodel.py │ │ │ ├── upscaling.py │ │ │ └── util.py │ │ ├── distributions │ │ │ ├── __init__.py │ │ │ └── distributions.py │ │ ├── ema.py │ │ ├── encoders │ │ │ ├── __init__.py │ │ │ └── modules.py │ │ ├── image_degradation │ │ │ ├── __init__.py │ │ │ ├── bsrgan.py │ │ │ ├── bsrgan_light.py │ │ │ ├── utils │ │ │ │ └── test.png │ │ │ └── utils_image.py │ │ └── midas │ │ │ ├── __init__.py │ │ │ ├── api.py │ │ │ ├── midas │ │ │ ├── __init__.py │ │ │ ├── base_model.py │ │ │ ├── blocks.py │ │ │ ├── dpt_depth.py │ │ │ ├── midas_net.py │ │ │ ├── midas_net_custom.py │ │ │ ├── transforms.py │ │ │ └── vit.py │ │ │ └── utils.py │ └── util.py └── stable_diffusion_inference │ ├── __init__.py │ ├── cloud.py │ ├── configs │ └── stable-diffusion │ │ ├── v1-inference.yaml │ │ ├── v2-inference-v.yaml │ │ ├── v2-inference.yaml │ │ ├── v2-inpainting-inference.yaml │ │ ├── v2-midas-inference.yaml │ │ └── x4-upscaling.yaml │ ├── data.py │ ├── factory.py │ ├── lit_model.py │ └── model.py ├── test.py └── tests └── test_stable_diffusion_inference.py /LICENSE: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/LICENSE -------------------------------------------------------------------------------- /LICENSE-MODEL: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/LICENSE-MODEL -------------------------------------------------------------------------------- /MANIFEST.in: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/MANIFEST.in -------------------------------------------------------------------------------- /README.md: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/README.md -------------------------------------------------------------------------------- /app-examples/app.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/app-examples/app.py -------------------------------------------------------------------------------- /app-examples/sd-compare/requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/app-examples/sd-compare/requirements.txt -------------------------------------------------------------------------------- /app-examples/sd-compare/sd-compare.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/app-examples/sd-compare/sd-compare.py -------------------------------------------------------------------------------- /cats in hats.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/cats in hats.png -------------------------------------------------------------------------------- /demo.ipynb: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/demo.ipynb -------------------------------------------------------------------------------- /example.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/example.py -------------------------------------------------------------------------------- /requirements.txt: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/requirements.txt -------------------------------------------------------------------------------- /setup.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/setup.py -------------------------------------------------------------------------------- /src/ldm1/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/lr_scheduler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/lr_scheduler.py -------------------------------------------------------------------------------- /src/ldm1/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/models/autoencoder.py -------------------------------------------------------------------------------- /src/ldm1/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/models/diffusion/classifier.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/models/diffusion/classifier.py -------------------------------------------------------------------------------- /src/ldm1/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/models/diffusion/ddim.py -------------------------------------------------------------------------------- /src/ldm1/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /src/ldm1/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/models/diffusion/plms.py -------------------------------------------------------------------------------- /src/ldm1/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/attention.py -------------------------------------------------------------------------------- /src/ldm1/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /src/ldm1/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /src/ldm1/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /src/ldm1/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/distributions/distributions.py -------------------------------------------------------------------------------- /src/ldm1/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/ema.py -------------------------------------------------------------------------------- /src/ldm1/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm1/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/encoders/modules.py -------------------------------------------------------------------------------- /src/ldm1/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /src/ldm1/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /src/ldm1/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /src/ldm1/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/image_degradation/utils/test.png -------------------------------------------------------------------------------- /src/ldm1/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /src/ldm1/modules/losses/__init__.py: -------------------------------------------------------------------------------- 1 | from ldm2.modules.losses.contperceptual import LPIPSWithDiscriminator -------------------------------------------------------------------------------- /src/ldm1/modules/losses/contperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/losses/contperceptual.py -------------------------------------------------------------------------------- /src/ldm1/modules/losses/vqperceptual.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/losses/vqperceptual.py -------------------------------------------------------------------------------- /src/ldm1/modules/x_transformer.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/modules/x_transformer.py -------------------------------------------------------------------------------- /src/ldm1/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm1/util.py -------------------------------------------------------------------------------- /src/ldm2/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/__init__.py -------------------------------------------------------------------------------- /src/ldm2/models/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/models/autoencoder.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/autoencoder.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/ddim.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/ddim.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/ddpm.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/ddpm.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/dpm_solver/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/dpm_solver/__init__.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/dpm_solver/dpm_solver.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/dpm_solver/dpm_solver.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/dpm_solver/sampler.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/dpm_solver/sampler.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/plms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/plms.py -------------------------------------------------------------------------------- /src/ldm2/models/diffusion/sampling_util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/models/diffusion/sampling_util.py -------------------------------------------------------------------------------- /src/ldm2/modules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/attention.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/attention.py -------------------------------------------------------------------------------- /src/ldm2/modules/diffusionmodules/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/diffusionmodules/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/diffusionmodules/model.py -------------------------------------------------------------------------------- /src/ldm2/modules/diffusionmodules/openaimodel.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/diffusionmodules/openaimodel.py -------------------------------------------------------------------------------- /src/ldm2/modules/diffusionmodules/upscaling.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/diffusionmodules/upscaling.py -------------------------------------------------------------------------------- /src/ldm2/modules/diffusionmodules/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/diffusionmodules/util.py -------------------------------------------------------------------------------- /src/ldm2/modules/distributions/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/distributions/distributions.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/distributions/distributions.py -------------------------------------------------------------------------------- /src/ldm2/modules/ema.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/ema.py -------------------------------------------------------------------------------- /src/ldm2/modules/encoders/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/encoders/modules.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/encoders/modules.py -------------------------------------------------------------------------------- /src/ldm2/modules/image_degradation/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/image_degradation/__init__.py -------------------------------------------------------------------------------- /src/ldm2/modules/image_degradation/bsrgan.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/image_degradation/bsrgan.py -------------------------------------------------------------------------------- /src/ldm2/modules/image_degradation/bsrgan_light.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/image_degradation/bsrgan_light.py -------------------------------------------------------------------------------- /src/ldm2/modules/image_degradation/utils/test.png: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/image_degradation/utils_image.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/image_degradation/utils_image.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/midas/api.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/api.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/__init__.py: -------------------------------------------------------------------------------- 1 | -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/base_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/base_model.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/blocks.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/blocks.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/dpt_depth.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/dpt_depth.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/midas_net.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/midas_net.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/midas_net_custom.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/midas_net_custom.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/transforms.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/transforms.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/midas/vit.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/midas/vit.py -------------------------------------------------------------------------------- /src/ldm2/modules/midas/utils.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/modules/midas/utils.py -------------------------------------------------------------------------------- /src/ldm2/util.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/ldm2/util.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/__init__.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/__init__.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/cloud.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/cloud.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/v1-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/v1-inference.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/v2-inference-v.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/v2-inference-v.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/v2-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/v2-inference.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/v2-inpainting-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/v2-inpainting-inference.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/v2-midas-inference.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/v2-midas-inference.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/configs/stable-diffusion/x4-upscaling.yaml: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/configs/stable-diffusion/x4-upscaling.yaml -------------------------------------------------------------------------------- /src/stable_diffusion_inference/data.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/data.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/factory.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/factory.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/lit_model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/lit_model.py -------------------------------------------------------------------------------- /src/stable_diffusion_inference/model.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/src/stable_diffusion_inference/model.py -------------------------------------------------------------------------------- /test.py: -------------------------------------------------------------------------------- 1 | import os 2 | -------------------------------------------------------------------------------- /tests/test_stable_diffusion_inference.py: -------------------------------------------------------------------------------- https://raw.githubusercontent.com/yangyuke001/SD-inference/HEAD/tests/test_stable_diffusion_inference.py --------------------------------------------------------------------------------