增加chatGLM int4配置支持 小显存也可以选择chatGLM

这个提交包含在:
fenglui
2023-07-22 08:29:15 +08:00
父节点 a393edfaa4
当前提交 5813d65e52
共有 2 个文件被更改,包括 8 次插入3 次删除

查看文件

@@ -80,6 +80,7 @@ ChatGLM_PTUNING_CHECKPOINT = "" # 例如"/home/hmp/ChatGLM2-6B/ptuning/output/6b
# 本地LLM模型如ChatGLM的执行方式 CPU/GPU
LOCAL_MODEL_DEVICE = "cpu" # 可选 "cuda"
LOCAL_MODEL_QUANT = "INT4" # 默认 "" "INT4" 启用量化INT4版本 "INT8" 启用量化INT8版本
# 设置gradio的并行线程数不需要修改