初始化项目,由ModelHub XC社区提供模型
Model: AI-ModelScope/Yi-1.5-34B-Chat-AWQ Source: Original Platform
This commit is contained in:
45
README.md
Normal file
45
README.md
Normal file
@@ -0,0 +1,45 @@
|
||||
---
|
||||
---
|
||||
language:
|
||||
- en
|
||||
- zh
|
||||
pipeline_tag: text-generation
|
||||
tags:
|
||||
- Awq
|
||||
- int4
|
||||
- yi1.5-34B-Chat
|
||||
- pytorch
|
||||
license: Apache 2.0
|
||||
license_name: Apache 2.0
|
||||
license_link: LICENSE
|
||||
---
|
||||
|
||||
## About Quantization
|
||||
我们使用modelscope [swift](https://github.com/modelscope/swift/)仓库进行AWQ 4bit量化. 量化文档可以查看[这里](https://github.com/modelscope/swift/blob/main/docs/source/LLM/LLM%E9%87%8F%E5%8C%96%E6%96%87%E6%A1%A3.md). 量化命令如下:
|
||||
|
||||
We use the modelscope [swift](https://github.com/modelscope/swift/) repository to perform AWQ 4bit quantization. Quantization documentation can be found [here](https://github.com/modelscope/swift/blob/main/docs/source_en/LLM/LLM-quantization.md). The quantization command is as follows:
|
||||
|
||||
```bash
|
||||
# Experimental Environment: A100
|
||||
swift export \
|
||||
--quant_bits 4 \
|
||||
--model_type yi-1_5-34b-chat \
|
||||
--quant_method awq \
|
||||
--quant_n_samples 32 \
|
||||
--dataset alpaca-zh alpaca-en sharegpt-gpt4-mini \
|
||||
--quant_seqlen 4096
|
||||
```
|
||||
|
||||
Inference:
|
||||
```bash
|
||||
CUDA_VISIBLE_DEVICES=0 swift infer --model_type yi-1_5-34b-chat-awq-int4
|
||||
```
|
||||
|
||||
SFT:
|
||||
```bash
|
||||
CUDA_VISIBLE_DEVICES=0 swift sft --model_type yi-1_5-34b-chat-awq-int4 --dataset leetcode-python-en
|
||||
```
|
||||
|
||||
Original Model:
|
||||
|
||||
[YI1.5-34B-Chat](https://modelscope.cn/models/01ai/Yi-1.5-34B-Chat/summary)
|
||||
Reference in New Issue
Block a user