初始化项目,由ModelHub XC社区提供模型
Model: ayh015/myLightningOPD Source: Original Platform
This commit is contained in:
57
setup.py
Normal file
57
setup.py
Normal file
@@ -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},
|
||||
)
|
||||
Reference in New Issue
Block a user