From a14463004c2754fef22675bac5ac226f502dfd07 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Mon, 30 Jun 2025 19:35:59 -0700 Subject: [PATCH] Prints --- pyproject.toml | 4 ++-- setup.py | 3 +++ 2 files changed, 5 insertions(+), 2 deletions(-) diff --git a/pyproject.toml b/pyproject.toml index 717f88f63a..28b94e5739 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -6,9 +6,9 @@ requires = [ "packaging>=24.2", "setuptools>=77.0.3,<80.0.0", "setuptools-scm>=8.0", - "torch<=2.7.0", "wheel", - "jinja2" + "jinja2", + "torch<=2.7.0", ] build-backend = "setuptools.build_meta" diff --git a/setup.py b/setup.py index 469af43cdd..4ada31180a 100644 --- a/setup.py +++ b/setup.py @@ -25,6 +25,7 @@ from setuptools.command.install import install # This arg is for multi-device UNSLOTH_TARGET_DEVICE = os.environ.get('UNSLOTH_TARGET_DEVICE', 'cuda') +print("[1] Installing Unsloth...") def load_module_from_path(module_name, path): spec = importlib.util.spec_from_file_location(module_name, path) @@ -40,6 +41,8 @@ ROOT_DIR = Path(__file__).parent # which is not installed yet ver = load_module_from_path('ver', os.path.join(ROOT_DIR, 'unsloth', 'version.py')) +print("[2] Installing Unsloth...") + def _is_cuda() -> bool: has_cuda = torch.version.cuda is not None return UNSLOTH_TARGET_DEVICE == "cuda" and has_cuda