Spaces:
Runtime error
Runtime error
File size: 1,778 Bytes
c3be39e |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 |
from pathlib import Path
from setuptools import find_packages, setup
if __name__ == "__main__":
with Path(Path(__file__).parent, "README.md").open(encoding="utf-8") as file:
long_description = file.read()
# TODO: This is a hack to get around the fact that we can't read the requirements.txt file, we should fix this.
# def _read_reqs(relpath):
# fullpath = os.path.join(Path(__file__).parent, relpath)
# with open(fullpath) as f:
# return [
# s.strip()
# for s in f.readlines()
# if (s.strip() and not s.startswith("#"))
# ]
REQUIREMENTS = [
"einops",
"einops-exts",
"transformers",
"torch",
"torchvision",
"pillow",
"more-itertools",
"datasets",
"braceexpand",
"webdataset",
"wandb",
"nltk",
"scipy",
"inflection",
"sentencepiece",
"open_clip_torch",
]
setup(
name="open_flamingo",
packages=find_packages(),
include_package_data=True,
version="0.0.2",
license="MIT",
description="An open-source framework for training large multimodal models",
long_description=long_description,
long_description_content_type="text/markdown",
data_files=[(".", ["README.md"])],
keywords=["machine learning"],
install_requires=REQUIREMENTS,
classifiers=[
"Development Status :: 4 - Beta",
"Intended Audience :: Developers",
"Topic :: Scientific/Engineering :: Artificial Intelligence",
"License :: OSI Approved :: MIT License",
"Programming Language :: Python :: 3.9",
],
)
|