在本地GPU上使用ExLlamaV2进行高效LLM推理-优快云博客

本文链接：https://blog.youkuaiyun.com/GEAWfaacc/article/details/145314034

随着大语言模型（LLM）的发展，越来越多开发者希望在本地硬件上运行这些模型。ExLlamaV2是一个高效的推理库，旨在利用现代消费级GPU来运行LLM。它特别适合使用GPTQ和EXL2量化模型。本篇文章将指导您如何在LangChain框架中使用ExLlamaV2运行这些模型。

技术背景介绍

ExLlamaV2设计目的是在本地硬件上高效地运行大语言模型，支持使用Hugging Face上的GPTQ和EXL2量化模型。相比直接在云端使用，这种方法能够有效减少延迟并降低成本，适合那些对隐私和实时性要求较高的应用场景。

核心原理解析

ExLlamaV2利用量化技术（如GPTQ）减少模型的内存占用，使得大模型可以在相对较小的GPU上运行。此外，它支持高性能的推理算法，这对于边缘计算和本地部署场景尤为重要。

代码实现演示

下面的代码展示了如何使用ExLlamaV2在LangChain中运行一个量化的LLM。

import os
from huggingface_hub import snapshot_download
from langchain_community.llms.exllamav2 import ExLlamaV2
from langchain_core.callbacks import StreamingStdOutCallbackHandler
from langchain_core.prompts import PromptTemplate
from libs.langchain.langchain.chains.llm import LLMChain
from exllamav2.generator import ExLlamaV2Sampler
import gc
import torch

# 用于下载GPTQ模型的函数
def download_GPTQ_model(model_name: str, models_dir: str = "./models/") -> str:
    if not os.path.exists(models_dir):
        os.makedirs(models_dir)

    _model_name = model_name.split("/")
    _model_name = "_".join(_model_name)
    model_path = os.path.join(models_dir, _model_name)
    if _model_name not in os.listdir(models_dir):
        snapshot_download(
            repo_id=model_name, local_dir=model_path, local_dir_use_symlinks=False
        )
    else:
        print(f"{model_name} already存在于目录中")

    return model_path

# 配置模型的推理设置
settings = ExLlamaV2Sampler.Settings()
settings.temperature = 0.85
settings.top_k = 50
settings.top_p = 0.8
settings.token_repetition_penalty = 1.05

# 下载模型
model_path = download_GPTQ_model("TheBloke/Mistral-7B-Instruct-v0.2-GPTQ")

# 设置回调处理
callbacks = [StreamingStdOutCallbackHandler()]

# 设置提示模板
template = """Question: {question}

Answer: Let's think step by step."""
prompt = PromptTemplate(template=template, input_variables=["question"])

# 初始化LLM
llm = ExLlamaV2(
    model_path=model_path,
    callbacks=callbacks,
    verbose=True,
    settings=settings,
    streaming=True,
    max_new_tokens=150,
)

# LLM链设置
llm_chain = LLMChain(prompt=prompt, llm=llm)

# 提出问题并生成答案
question = "What Football team won the UEFA Champions League in the year the iphone 6s was released?"
output = llm_chain.invoke({"question": question})
print(output)

# 清理缓存
torch.cuda.empty_cache()
gc.collect()