Update services/llama_generator.py
Browse files
services/llama_generator.py
CHANGED
@@ -50,7 +50,7 @@ class LlamaGenerator(BaseGenerator):
|
|
50 |
|
51 |
self.model_manager = ModelManager()
|
52 |
|
53 |
-
self.tokenizer = model_manager.load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
|
54 |
|
55 |
super().__init__(
|
56 |
llama_model_name,
|
|
|
50 |
|
51 |
self.model_manager = ModelManager()
|
52 |
|
53 |
+
self.tokenizer = self.model_manager.load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
|
54 |
|
55 |
super().__init__(
|
56 |
llama_model_name,
|