ColossalAI/examples/images/diffusion/setup.py

from setuptools import find_packages, setup

setup(
    name="latent-diffusion",
    version="0.0.1",
    description="",
    packages=find_packages(),
    install_requires=[
        "torch",
        "numpy",
        "tqdm",
    ],
)