| from setuptools import setup, find_packages | |
| setup( | |
| name="cwm", | |
| version="1.0", | |
| packages=find_packages(), | |
| description="Video modeling transformers package for MAE", | |
| author="Dan Bear, Kevin Figelis, Honglin Chen, Rahul Venakatesh, Wanhee Lee, Dan Yamins, Klemen Kotar", | |
| install_requires=[ | |
| # 'scipy', | |
| # 'scikit-learn', | |
| # 'matplotlib', | |
| # 'h5py', | |
| # 'kornia', | |
| # 'future', | |
| # 'einops', | |
| # 'timm', | |
| # 'opencv-python', | |
| # 'decord', | |
| # 'iopath', | |
| # 'pandas', | |
| # 'IPython', | |
| # 'tensorboardx', | |
| # 'tensorboard', | |
| # 'positional_encodings', | |
| # 'scikit-image', | |
| # 'packaging', | |
| # 'wandb', | |
| # 'numpy==1.23.5', | |
| # 'torch==2.0.0', | |
| # 'torchvision==0.15.1' | |
| ], | |
| ) | |