xusenlinzy / api-for-open-llm

Openai style api for open large language models, using LLMs just as chatgpt! Support for LLaMA, LLaMA-2, BLOOM, Falcon, Baichuan, Qwen, Xverse, SqlCoder, CodeLLaMA, ChatGLM, ChatGLM2, ChatGLM3 etc. 开源大模型的统一后端接口
Apache License 2.0
2.36k stars 270 forks source link

RuntimeError: CUDA error: device-side assert triggered #302

Closed ChaoPeng13 closed 3 months ago

ChaoPeng13 commented 3 months ago

提交前必须检查以下项目 | The following items must be checked before submission

问题类型 | Type of problem

模型推理和部署 | Model inference and deployment

操作系统 | Operating system

Windows

详细描述问题 | Detailed description of the problem

The .env file is below

PORT=8000

# model related
MODEL_NAME=chatglm3
MODEL_PATH=D:\Projects\LLMModels\chatglm3-6b
CONTEXT_LEN=
LOAD_IN_8BIT=false
LOAD_IN_4BIT=false
PROMPT_NAME=chatglm3

# rag related
EMBEDDING_NAME=
RERANK_NAME=

# device related
# "auto", "cuda:0", "cuda:1", ...
DEVICE_MAP=cuda:0
GPUS=
NUM_GPUs=1
DTYPE=half

# api related
API_PREFIX=/v1

USE_STREAMER_V2=false
ENGINE=default

TASKS=llm
# TASKS=llm,rag

When tried to chat with chatclm3, I got " RuntimeError: CUDA error: device-side assert triggered"

Dependencies

No response

运行日志或截图 | Runtime logs or screenshots

No response