From 2b50604876b4c15cd809f7b8a7d3b8c22b5b22a7 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 30 Jun 2025 19:50:15 -0700 Subject: [PATCH] Update setup.py --- setup.py | 6 +++--- 1 file changed, 3 insertions(+), 3 deletions(-) diff --git a/setup.py b/setup.py index d891eeedce..e064511988 100644 --- a/setup.py +++ b/setup.py @@ -13,12 +13,12 @@ from pathlib import Path from shutil import which import shutil -import torch +# import torch from packaging.version import Version, parse from setuptools import Extension, setup from setuptools.command.build_ext import build_ext from setuptools_scm import get_version -from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME +# from torch.utils.cpp_extension import CUDA_HOME, ROCM_HOME from setuptools.command.install import install @@ -838,7 +838,7 @@ if _is_hip(): } print(get_unsloth_version()) -print(get_requirements()) +# print(get_requirements()) setup( # static metadata should rather go in pyproject.toml version=get_unsloth_version(),