Upload setup.py with huggingface_hub

setup.py

@@ -0,0 +1,57 @@
+# SPDX-FileCopyrightText: Copyright (c) 2026 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
+# SPDX-License-Identifier: Apache-2.0
+
+import sys
+import platform
+
+from setuptools import find_packages, setup
+from wheel.bdist_wheel import bdist_wheel as _bdist_wheel
+
+
+def _fetch_requirements(path):
+    with open(path, "r") as fd:
+        return [r.strip() for r in fd.readlines() if r.strip() and not r.startswith("#")]
+
+
+# Custom wheel class to modify the wheel name
+class bdist_wheel(_bdist_wheel):
+    def finalize_options(self):
+        _bdist_wheel.finalize_options(self)
+        self.root_is_pure = False
+
+    def get_tag(self):
+        python_version = f"cp{sys.version_info.major}{sys.version_info.minor}"
+        abi_tag = f"{python_version}"
+
+        if platform.system() == "Linux":
+            platform_tag = "manylinux1_x86_64"
+        else:
+            platform_tag = platform.system().lower()
+
+        return python_version, abi_tag, platform_tag
+
+
+# Setup configuration
+setup(
+    author="slime Team",
+    name="slime",
+    version="0.1.0",
+    packages=find_packages(include=["slime*", "slime_plugins*"]),
+    include_package_data=True,
+    install_requires=_fetch_requirements("requirements.txt"),
+    extras_require={
+        "fsdp": [
+            "torch>=2.0",
+        ]
+    },
+    python_requires=">=3.10",
+    classifiers=[
+        "Programming Language :: Python :: 3.10",
+        "Programming Language :: Python :: 3.11",
+        "Programming Language :: Python :: 3.12",
+        "Environment :: GPU :: NVIDIA CUDA",
+        "Topic :: Scientific/Engineering :: Artificial Intelligence",
+        "Topic :: System :: Distributed Computing",
+    ],
+    cmdclass={"bdist_wheel": bdist_wheel},
+)