From 8bb3483cdf81b289ac093152045095133114613d Mon Sep 17 00:00:00 2001 From: Brian Vaughan Date: Mon, 19 Feb 2024 16:42:58 -0500 Subject: [PATCH 1/2] setup.py to install so we can add tests and move scripts --- .github/workflows/mypy.yml | 2 +- setup.py | 17 +++++++++++++++++ 2 files changed, 18 insertions(+), 1 deletion(-) create mode 100644 setup.py diff --git a/.github/workflows/mypy.yml b/.github/workflows/mypy.yml index 6081a268..b9e6c210 100644 --- a/.github/workflows/mypy.yml +++ b/.github/workflows/mypy.yml @@ -50,7 +50,7 @@ jobs: pip install -I ibm-fms@git+https://github.com/foundation-model-stack/foundation-model-stack@main # No type stubs available for "fire" and "transformers" - mypy --exclude fms_to_hf.py --exclude main_training.py . + mypy --exclude fms_to_hf.py --exclude main_training.py --exclude setup.py . - name: Save Virtualenv id: cache-venv-save diff --git a/setup.py b/setup.py new file mode 100644 index 00000000..2939b5a1 --- /dev/null +++ b/setup.py @@ -0,0 +1,17 @@ +from setuptools import find_packages, setup + +setup( + name="fms_fsdp", + version="0.0.1", + author="Linsong Chu, Davis Wertheimer, Brian Vaughan, Andrea Frittoli, Joshua Rosenkranz, Antoni Viros i Martin, Raghu Kiran Ganti", + author_email="lchu@us.ibm.com", + description="Pretraining scripts using FSDP and IBM Foundation Model Stack", + url="https://github.com/foundation-model-stack/fms-fsdp", + packages=find_packages(), + install_requires=["ibm-fms >= 0.0.3", "torch >= 2.1"], + license="Apache License 2.0", + classifiers=[ + "Programming Language :: Python :: 3", + "License :: OSI Approved :: Apache Software License", + ], +) From 507774f727ee8e68b9f9c5e96f179ad3792ef404 Mon Sep 17 00:00:00 2001 From: Brian Vaughan Date: Mon, 19 Feb 2024 16:55:18 -0500 Subject: [PATCH 2/2] isort --- setup.py | 1 + 1 file changed, 1 insertion(+) diff --git a/setup.py b/setup.py index 2939b5a1..d93301ab 100644 --- a/setup.py +++ b/setup.py @@ -1,5 +1,6 @@ from setuptools import find_packages, setup + setup( name="fms_fsdp", version="0.0.1",