Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -20,7 +20,7 @@ model = LlamaForCausalLM.from_pretrained(
|
|
20 |
|
21 |
MAX_SEQ_LEN = 4096
|
22 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
23 |
-
model.
|
24 |
|
25 |
|
26 |
|
|
|
20 |
|
21 |
MAX_SEQ_LEN = 4096
|
22 |
tokenizer = AutoTokenizer.from_pretrained(config.base_model_name_or_path)
|
23 |
+
model.pad_token = tokenizer.pad_token = tokenizer.unk_token
|
24 |
|
25 |
|
26 |
|