From 32347901d4af94ccd72b3c7e1afaaceb5cb3d26a Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Sun, 28 Apr 2024 03:49:13 +0800
Subject: [PATCH] fix setup

---
 README.md    | 2 +-
 README_zh.md | 2 +-
 setup.py     | 1 -
 3 files changed, 2 insertions(+), 3 deletions(-)

diff --git a/README.md b/README.md
index 789c20ce..04e5aa5b 100644
--- a/README.md
+++ b/README.md
@@ -322,7 +322,7 @@ cd LLaMA-Factory
 pip install -e .[metrics]
 ```
 
-Extra dependencies available: deepspeed, metrics, unsloth, galore, badam, vllm, bitsandbytes, gptq, awq, aqlm, qwen, modelscope, quality
+Extra dependencies available: deepspeed, metrics, galore, badam, vllm, bitsandbytes, gptq, awq, aqlm, qwen, modelscope, quality
 
 <details><summary>For Windows users</summary>
 
diff --git a/README_zh.md b/README_zh.md
index fb02aa95..2240c688 100644
--- a/README_zh.md
+++ b/README_zh.md
@@ -322,7 +322,7 @@ cd LLaMA-Factory
 pip install -e .[metrics]
 ```
 
-可选的额外依赖项：deepspeed、metrics、unsloth、galore、badam、vllm、bitsandbytes、gptq、awq、aqlm、qwen、modelscope、quality
+可选的额外依赖项：deepspeed、metrics、galore、badam、vllm、bitsandbytes、gptq、awq、aqlm、qwen、modelscope、quality
 
 <details><summary>Windows 用户指南</summary>
 
diff --git a/setup.py b/setup.py
index 7ff3185f..6a03138d 100644
--- a/setup.py
+++ b/setup.py
@@ -22,7 +22,6 @@ def get_requires():
 extra_require = {
     "deepspeed": ["deepspeed>=0.10.0"],
     "metrics": ["nltk", "jieba", "rouge-chinese"],
-    "unsloth": ["torch==2.2.0", "unsloth[cu121-ampere-torch220]"],
     "galore": ["galore-torch"],
     "badam": ["badam"],
     "vllm": ["vllm>=0.4.0"],