# Copyright (c) Facebook, Inc. and its affiliates. All Rights Reserved.

from setuptools import find_packages, setup

setup(
    name="timesformer",
    version="1.0",
    author="FBAI",
    url="unknown",
    description="TimeSformer",
    keywords = [
    'artificial intelligence',
    'attention mechanism',
    'transformers',
    'video classification',
    ],
    install_requires=[
        'einops>=0.3',
        'torch>=1.6'
    ],
    extras_require={"tensorboard_video_visualization": ["moviepy"]},
    packages=find_packages(exclude=("configs", "tests")),
)