ColossalAI/setup.py at 1fa8c5e09ff7422c30fe7683beb209bfba7e153b - ColossalAI - https://git.xinac.net

Making large AI models cheaper, faster and more accessible

ai big-model data-parallelism deep-learning distributed-computing foundation-models heterogeneous-training hpc inference large-scale model-parallelism pipeline-parallelism

13 lines

234 B

Raw Blame History

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