File size: 622 Bytes
9c3a994 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 |
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="[email protected]",
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",
],
)
|