Update services/model_service.py
Browse files
services/model_service.py
CHANGED
@@ -25,7 +25,7 @@ class ModelService:
|
|
25 |
@lru_cache(maxsize=1)
|
26 |
def _load_models(self):
|
27 |
try:
|
28 |
-
self.tokenizer = AutoTokenizer.from_pretrained(settings.MODEL_NAME)
|
29 |
|
30 |
# Modify the model configuration to use a valid rope_scaling format
|
31 |
config = LlamaConfig.from_pretrained(settings.model_name)
|
|
|
25 |
@lru_cache(maxsize=1)
|
26 |
def _load_models(self):
|
27 |
try:
|
28 |
+
self.tokenizer = AutoTokenizer.from_pretrained(settings.MODEL_NAME )
|
29 |
|
30 |
# Modify the model configuration to use a valid rope_scaling format
|
31 |
config = LlamaConfig.from_pretrained(settings.model_name)
|