Update app.py
Browse files
app.py
CHANGED
@@ -418,7 +418,7 @@ class LlamaGenerator(BaseGenerator):
|
|
418 |
|
419 |
):
|
420 |
|
421 |
-
|
422 |
|
423 |
def load_model(self, model_name: str):
|
424 |
# Code to load your model, e.g., Hugging Face's transformers library
|
|
|
418 |
|
419 |
):
|
420 |
|
421 |
+
self.tokenizer = self.load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
|
422 |
|
423 |
def load_model(self, model_name: str):
|
424 |
# Code to load your model, e.g., Hugging Face's transformers library
|