From 5cd4947650403490419e5bddf2b1ac7e137edf8b Mon Sep 17 00:00:00 2001
From: hiyouga <hiyouga@buaa.edu.cn>
Date: Fri, 8 Mar 2024 01:23:00 +0800
Subject: [PATCH] Update setup.py

---
 setup.py | 5 +++--
 1 file changed, 3 insertions(+), 2 deletions(-)

diff --git a/setup.py b/setup.py
index 412dd084..ca8db70a 100644
--- a/setup.py
+++ b/setup.py
@@ -21,13 +21,14 @@ def get_requires():
 extra_require = {
     "deepspeed": ["deepspeed==0.13.1"],
     "metrics": ["nltk", "jieba", "rouge-chinese"],
-    "unsloth": ["unsloth[cu121-ampere-torch220] @ git+https://github.com/unslothai/unsloth.git"],
+    "unsloth": ["torch==2.2.0", "unsloth[cu121-ampere-torch220] @ git+https://github.com/unslothai/unsloth.git"],
     "vllm": ["vllm==0.3.3"],
     "bitsandbytes": ["bitsandbytes>=0.39.0"],
     "gptq": ["optimum>=1.16.0", "auto-gptq>=0.5.0"],
     "awq": ["autoawq"],
     "aqlm": ["aqlm[gpu,cpu]"],
-    "galore": ["galore_torch @ git+https://github.com/jiaweizzhao/GaLore.git"],
+    "galore": ["galore_torch @ git+https://github.com/hiyouga/GaLore.git"],
+    "qwen": ["tiktoken", "transformers_stream_generator"],
 }