Update app.py
Browse files
app.py
CHANGED
@@ -446,9 +446,7 @@ class LlamaGenerator(BaseGenerator):
|
|
446 |
return AutoTokenizer.from_pretrained(model_name)
|
447 |
|
448 |
self.tokenizer = load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
|
449 |
-
|
450 |
-
|
451 |
-
@observe()
|
452 |
super().__init__(
|
453 |
llama_model_name,
|
454 |
device,
|
|
|
446 |
return AutoTokenizer.from_pretrained(model_name)
|
447 |
|
448 |
self.tokenizer = load_tokenizer(llama_model_name) # Add this line to initialize the tokenizer
|
449 |
+
|
|
|
|
|
450 |
super().__init__(
|
451 |
llama_model_name,
|
452 |
device,
|