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',
  ],
)