增加requirements.txt

README.md

@@ -14,16 +14,20 @@ docker pull image.sourcefind.cn:5000/dcu/admin/base/pytorch:2.1.0-centos7.6-dtk2
 # <Image ID>用上面拉取docker镜像的ID替换
 # <Host Path>主机端路径
 # <Container Path>容器映射路径
-docker run -it --name baichuan --shm-size=1024G -v /opt/hyhal:/opt/hyhal:ro --device=/dev/kfd --device=/dev/dri/ --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v <Host Path>:<Container Path> <Image ID> /bin/bash
+docker run -it --name mydocker --shm-size=1024G -v /opt/hyhal:/opt/hyhal:ro --device=/dev/kfd --device=/dev/dri/ --cap-add=SYS_PTRACE --security-opt seccomp=unconfined --ulimit memlock=-1:-1 --ipc=host --network host --group-add video -v <Host Path>:<Container Path> <Image ID> /bin/bash                  
 ```
-注：
-1、docker启动  -v /opt/hyhal:/opt/hyhal  这个变量不能少             
+注：                      
+1、docker启动  -v /opt/hyhal:/opt/hyhal  这个变量不能少                                
+2、若使用 pip install 下载安装过慢，可添加源：-i https://pypi.tuna.tsinghua.edu.cn/simple/                              
 
 #### 源码编译安装
 - 代码下载
 根据不同的需求下载不同的分支
 - 提供2种源码编译方式（进入AutoAWQ目录）：
 ```
+#基础依赖安装：                      
+pip install -r requirements.txt  
+
 1. 源码编译安装
 pip3 install e.
 
@@ -32,7 +36,29 @@ pip3 install e.
 python3 setup.py bdist_wheel
 cd dist && pip3 install autoawq*
 ```
-
+## 支持模型
+| Models   | Sizes                       |
+| :------: | :-------------------------: |
+| LLaMA-2  | 7B/13B/70B                  |
+| LLaMA    | 7B/13B/30B/65B              |
+| Mistral  | 7B                          |
+| Vicuna   | 7B/13B                      |
+| MPT      | 7B/30B                      |
+| Falcon   | 7B/40B                      |
+| OPT      | 125m/1.3B/2.7B/6.7B/13B/30B |
+| Bloom    | 560m/3B/7B/                 |
+| GPTJ     | 6.7B                        |
+| Aquila   | 7B                          |
+| Aquila2  | 7B/34B                      |
+| Yi       | 6B/34B                      |
+| Qwen     | 1.8B/7B/14B/72B             |
+| BigCode  | 1B/7B/15B                   |
+| GPT NeoX | 20B                         |
+| GPT-J    | 6B                          |
+| LLaVa    | 7B/13B                      |
+| Mixtral  | 8x7B                        |
+| Baichuan | 7B/13B                      |
+| QWen     | 1.8B/7B/14/72B              |
 
 
 

requirements.txt

+zstandard
+transformers==4.42.3
\ No newline at end of file