| from setuptools import setup, find_packages |
| from torch.utils.cpp_extension import BuildExtension, CUDAExtension |
| from distutils.extension import Extension |
| from Cython.Build import cythonize |
| import numpy as np |
|
|
|
|
| setup( |
| name="michelangelo", |
| version="0.4.1", |
| author="Zibo Zhao, Wen Liu and Xin Chen", |
| author_email="stevenxxliu@tencent.com", |
| description="Michelangelo: a 3D Shape Generation System.", |
| packages=find_packages(exclude=("configs", "tests", "scripts", "deps")), |
| python_requires=">=3.8", |
| install_requires=[ |
| "torch", |
| "numpy", |
| "cython", |
| "tqdm", |
| ], |
| ) |
|
|