| from setuptools import setup, find_packages | |
| setup( | |
| name = 'transformer_maskgit', | |
| install_requires=[ | |
| 'accelerate', | |
| 'beartype', | |
| 'einops>=0.6', | |
| 'ema-pytorch>=0.2.2', | |
| 'opencv-python', | |
| 'pillow', | |
| 'numpy', | |
| 'sentencepiece==0.2.0', | |
| #'torch==2.0.1', | |
| 'torchtyping', | |
| 'torchvision', | |
| #'transformers==4.30.1', | |
| 'tqdm', | |
| 'vector-quantize-pytorch==1.1.2', | |
| 'nibabel', | |
| 'openpyxl', | |
| ], | |
| ) | |