Spaces:
Sleeping
Sleeping
TTsamurai
commited on
Commit
·
9be606d
1
Parent(s):
dfbbb86
delete tokenizer
Browse files
app.py
CHANGED
|
@@ -821,8 +821,9 @@ if __name__ == "__main__":
|
|
| 821 |
|
| 822 |
personality_prompt = build_personality_prompt(personality)
|
| 823 |
system_instruction_without_context = SYSTEM_INSTRUCTION + "\n" + personality_prompt + "\n"
|
| 824 |
-
tokenizer = AutoTokenizer.from_pretrained(RESPONSE_GENERATOR)
|
| 825 |
-
tokenizer, terminator = prepare_tokenizer(tokenizer)
|
|
|
|
| 826 |
print(f"Terminator: {terminator}")
|
| 827 |
demo = create_demo(terminator, system_instruction_without_context, stock_context_list, raw_context_list)
|
| 828 |
demo.launch(share=True)
|
|
|
|
| 821 |
|
| 822 |
personality_prompt = build_personality_prompt(personality)
|
| 823 |
system_instruction_without_context = SYSTEM_INSTRUCTION + "\n" + personality_prompt + "\n"
|
| 824 |
+
# tokenizer = AutoTokenizer.from_pretrained(RESPONSE_GENERATOR)
|
| 825 |
+
# tokenizer, terminator = prepare_tokenizer(tokenizer)
|
| 826 |
+
terminator = ["<eos>", "<unk>", "<sep>", "<pad>", "<cls>", "<mask>"]
|
| 827 |
print(f"Terminator: {terminator}")
|
| 828 |
demo = create_demo(terminator, system_instruction_without_context, stock_context_list, raw_context_list)
|
| 829 |
demo.launch(share=True)
|