resolve python 3.8 package

Former-commit-id: 75aec4cf8e9089808a1731e2848b29191f26d51d
This commit is contained in:
hiyouga 2024-05-09 16:52:27 +08:00
parent 5ff89a0f32
commit 9dc209b458
5 changed files with 20 additions and 6 deletions

View File

@ -310,13 +310,19 @@ huggingface-cli login
### Installation ### Installation
> [!IMPORTANT]
> Installation is mandatory.
```bash ```bash
git clone https://github.com/hiyouga/LLaMA-Factory.git git clone https://github.com/hiyouga/LLaMA-Factory.git
cd LLaMA-Factory cd LLaMA-Factory
pip install -e .[metrics] pip install -e .[torch,metrics]
``` ```
Extra dependencies available: metrics, deepspeed, bitsandbytes, vllm, galore, badam, gptq, awq, aqlm, qwen, modelscope, quality Extra dependencies available: torch, metrics, deepspeed, bitsandbytes, vllm, galore, badam, gptq, awq, aqlm, qwen, modelscope, quality
> [!TIP]
> Use `pip install --no-deps -e .` to resolve package conflicts.
<details><summary>For Windows users</summary> <details><summary>For Windows users</summary>

View File

@ -310,13 +310,19 @@ huggingface-cli login
### 安装 LLaMA Factory ### 安装 LLaMA Factory
> [!IMPORTANT]
> 此步骤为必需。
```bash ```bash
git clone https://github.com/hiyouga/LLaMA-Factory.git git clone https://github.com/hiyouga/LLaMA-Factory.git
cd LLaMA-Factory cd LLaMA-Factory
pip install -e .[metrics] pip install -e .[torch,metrics]
``` ```
可选的额外依赖项metrics、deepspeed、bitsandbytes、vllm、galore、badam、gptq、awq、aqlm、qwen、modelscope、quality 可选的额外依赖项torch、metrics、deepspeed、bitsandbytes、vllm、galore、badam、gptq、awq、aqlm、qwen、modelscope、quality
> [!TIP]
> 遇到包冲突时,可使用 `pip install --no-deps -e .` 解决。
<details><summary>Windows 用户指南</summary> <details><summary>Windows 用户指南</summary>

View File

@ -1,4 +1,3 @@
torch>=1.13.1
transformers>=4.37.2 transformers>=4.37.2
datasets>=2.14.3 datasets>=2.14.3
accelerate>=0.27.2 accelerate>=0.27.2

View File

@ -20,6 +20,7 @@ def get_requires():
extra_require = { extra_require = {
"torch": ["torch>=1.13.1"],
"metrics": ["nltk", "jieba", "rouge-chinese"], "metrics": ["nltk", "jieba", "rouge-chinese"],
"deepspeed": ["deepspeed>=0.10.0,<=0.14.0"], "deepspeed": ["deepspeed>=0.10.0,<=0.14.0"],
"bitsandbytes": ["bitsandbytes>=0.39.0"], "bitsandbytes": ["bitsandbytes>=0.39.0"],

View File

@ -1,6 +1,8 @@
import os import os
from contextlib import asynccontextmanager from contextlib import asynccontextmanager
from typing import Annotated, Optional from typing import Optional
from typing_extensions import Annotated
from ..chat import ChatModel from ..chat import ChatModel
from ..extras.misc import torch_gc from ..extras.misc import torch_gc