Update vllm_support/src/kalm_t5gemma2_vllm_plugin/constants.py
Browse files
vllm_support/src/kalm_t5gemma2_vllm_plugin/constants.py
CHANGED
|
@@ -3,7 +3,7 @@ from __future__ import annotations
|
|
| 3 |
from typing import Iterable
|
| 4 |
|
| 5 |
|
| 6 |
-
MODEL_ID = "KaLM-Embedding/KaLM-Reranker-V1-
|
| 7 |
PLUGIN_NAME = "kalm_t5gemma2"
|
| 8 |
ARCHITECTURE = "T5Gemma2VllmScoreClassification"
|
| 9 |
TEXT_MODALITY = "text"
|
|
|
|
| 3 |
from typing import Iterable
|
| 4 |
|
| 5 |
|
| 6 |
+
MODEL_ID = "KaLM-Embedding/KaLM-Reranker-V1-Large"
|
| 7 |
PLUGIN_NAME = "kalm_t5gemma2"
|
| 8 |
ARCHITECTURE = "T5Gemma2VllmScoreClassification"
|
| 9 |
TEXT_MODALITY = "text"
|