Chris4K commited on
Commit
2c93520
·
verified ·
1 Parent(s): b5e231e

Update services/model_service.py

Browse files
Files changed (1) hide show
  1. services/model_service.py +1 -1
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)