diff --git a/api/internal/storeLink/storeLink.go b/api/internal/storeLink/storeLink.go index 2cd54f59..69cfc02d 100644 --- a/api/internal/storeLink/storeLink.go +++ b/api/internal/storeLink/storeLink.go @@ -78,7 +78,7 @@ var ( } ModelTypeMap = map[string][]string{ "image_recognition": {"imagenet_resnet50"}, - "text_to_text": {"chatGLM-6B"}, + "text_to_text": {"chatGLM_6B"}, } AITYPE = map[string]string{ "1": OCTOPUS,