Spaces:
Build error
Build error
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="mmgpt", | |
packages=find_packages(), | |
include_package_data=True, | |
version="0.0.1", | |
license="Apache 2.0", | |
description="An open-source framework for multi-modality instruction fine-tuning", | |
long_description=long_description, | |
long_description_content_type="text/markdown", | |
data_files=[(".", ["README.md"])], | |
keywords=["machine learning"], | |
install_requires=REQUIREMENTS, | |
) | |