perf: Optimized tensorrtLLM display performanceperf (#11184)

This commit is contained in:
CityFun 2025-12-04 17:57:19 +08:00 committed by GitHub
parent 4fb099912e
commit 7d25228e55
No known key found for this signature in database
GPG key ID: B5690EEEBB952194
2 changed files with 17 additions and 22 deletions

View file

@ -88,31 +88,25 @@ func (t TensorRTLLMService) Page(req request.TensorRTLLMSearch) response.TensorR
} }
composeByte, err := files.NewFileOp().GetContent(path.Join(global.Dir.TensorRTLLMDir, item.Name, "docker-compose.yml")) composeByte, err := files.NewFileOp().GetContent(path.Join(global.Dir.TensorRTLLMDir, item.Name, "docker-compose.yml"))
if err != nil { if err == nil {
continue serverDTO.Environments, _ = getDockerComposeEnvironments(composeByte)
}
serverDTO.Environments, err = getDockerComposeEnvironments(composeByte)
if err != nil {
continue
} }
volumes, err := getDockerComposeVolumes(composeByte) volumes, err := getDockerComposeVolumes(composeByte)
if err != nil { if err == nil {
continue var defaultVolumes = map[string]string{
} "${MODEL_PATH}": "${MODEL_PATH}",
var defaultVolumes = map[string]string{
"${MODEL_PATH}": "${MODEL_PATH}",
}
for _, volume := range volumes {
exist := false
for key, value := range defaultVolumes {
if key == volume.Source && value == volume.Target {
exist = true
break
}
} }
if !exist { for _, volume := range volumes {
serverDTO.Volumes = append(serverDTO.Volumes, volume) exist := false
for key, value := range defaultVolumes {
if key == volume.Source && value == volume.Target {
exist = true
break
}
}
if !exist {
serverDTO.Volumes = append(serverDTO.Volumes, volume)
}
} }
} }
items = append(items, serverDTO) items = append(items, serverDTO)

View file

@ -110,6 +110,7 @@ const openEdit = (rowData: any): void => {
if (tensorRTLLM.value.exposedPorts == null) { if (tensorRTLLM.value.exposedPorts == null) {
tensorRTLLM.value.exposedPorts = []; tensorRTLLM.value.exposedPorts = [];
} }
tensorRTLLM.value.command = rowData.command.slice(1, -1).replace(/^'|'$/g, '').replace(/\\"/g, '"');
if (tensorRTLLM.value.extraHosts == null) { if (tensorRTLLM.value.extraHosts == null) {
tensorRTLLM.value.extraHosts = []; tensorRTLLM.value.extraHosts = [];
} }