DB-GPT
DB-GPT copied to clipboard
[Bug] [Module Name] vLLM Start Qwen2.5-32B-Instruct Error
Search before asking
- [X] I had searched in the issues and found no similar issues.
Operating system information
Linux
Python version information
3.10
DB-GPT version
main
Related scenes
- [ ] Chat Data
- [ ] Chat Excel
- [ ] Chat DB
- [ ] Chat Knowledge
- [ ] Model Management
- [ ] Dashboard
- [ ] Plugins
Installation Information
-
[ ] AutoDL Image
-
[ ] Other
Device information
GPU A800
Models information
LLM: Qwen2.5-32B-Instruct
What happened
Traceback (most recent call last):
File "/usr/local/bin/dbgpt", line 8, in
What you expected to happen
def model_param_class(self, model_type: str = None) -> BaseModelParameters:
import argparse
from vllm.engine.arg_utils import AsyncEngineArgs
parser = argparse.ArgumentParser()
parser = AsyncEngineArgs.add_cli_args(parser)
parser.add_argument("--model_name", type=str, help="model name")
parser.add_argument(
"--model_path",
type=str,
help="local model path of the huggingface model to use",
)
parser.add_argument("--model_type", type=str, help="model type")
# parser.add_argument("--device", type=str, default=None, help="device")
# TODO parse prompt templete from `model_name` and `model_path`
parser.add_argument(
"--prompt_template",
type=str,
default=None,
help="Prompt template. If None, the prompt template is automatically determined from model path",
)
descs = _extract_parameter_details(
parser,
"dbgpt.model.parameter.VLLMModelParameters",
skip_names=["model"],
overwrite_default_values={"trust_remote_code": True},
)
return _build_parameter_class(descs)
dbgpt.model.parameter.VLLMModelParameters not find
How to reproduce
#!/bin/bash
dbgpt start worker
--model_type vllm
--model_name qwen2.5-32b-instruct
--model_path /app/models/Qwen2.5-32B-Instruct
--port 8005
--gpu_memory_utilization 0.92
--max_model_len 4096
--controller_addr http://10.28.92.172:8000
Additional context
source version 0.5.10
Are you willing to submit PR?
- [ ] Yes I am willing to submit a PR!