From f520bc39eef20e6ec7b9518dcee06290a3c171f3 Mon Sep 17 00:00:00 2001 From: Daniel Han-Chen Date: Thu, 30 Nov 2023 17:05:49 +1100 Subject: [PATCH] requirements --- pyproject.toml | 4 ++++ cu118.txt => requirements_cu118.txt | 0 cu121.txt => requirements_cu121.txt | 0 3 files changed, 4 insertions(+) rename cu118.txt => requirements_cu118.txt (100%) rename cu121.txt => requirements_cu121.txt (100%) diff --git a/pyproject.toml b/pyproject.toml index e61afb39c1..43089c6c4e 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -25,6 +25,10 @@ classifiers = [ [tool.setuptools.dynamic] version = {attr = "unsloth.__version__"} dependencies = {file = ["requirements.txt"]} +optional-dependencies = { + cu118 = { file = ["requirements_cu118.txt"] }, + cu121 = { file = ["requirements_cu121.txt"] }, +} [project.urls] homepage = "http://www.unsloth.ai" diff --git a/cu118.txt b/requirements_cu118.txt similarity index 100% rename from cu118.txt rename to requirements_cu118.txt diff --git a/cu121.txt b/requirements_cu121.txt similarity index 100% rename from cu121.txt rename to requirements_cu121.txt