Update services/model_service.py
Browse files
services/model_service.py
CHANGED
@@ -23,7 +23,7 @@ class ModelService:
|
|
23 |
def _load_models(self):
|
24 |
try:
|
25 |
# Load tokenizer
|
26 |
-
self.tokenizer = AutoTokenizer.from_pretrained(settings.MODEL_NAME)
|
27 |
|
28 |
## Load model configuration
|
29 |
#config = LlamaConfig.from_pretrained(settings.MODEL_NAME)
|
|
|
23 |
def _load_models(self):
|
24 |
try:
|
25 |
# Load tokenizer
|
26 |
+
#self.tokenizer = AutoTokenizer.from_pretrained(settings.MODEL_NAME)
|
27 |
|
28 |
## Load model configuration
|
29 |
#config = LlamaConfig.from_pretrained(settings.MODEL_NAME)
|