Spaces:
Running
Running
Update chatbot.py
Browse files- chatbot.py +1 -1
chatbot.py
CHANGED
@@ -24,7 +24,7 @@ class Comsatsbot:
|
|
24 |
# self.memory = ConversationTokenBufferMemory(llm=self.llm, max_token_limit=3000)
|
25 |
|
26 |
# Load LLaMA-compatible tokenizer
|
27 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
28 |
|
29 |
# Define a simple token counting function
|
30 |
def llama_token_count(text: str) -> int:
|
|
|
24 |
# self.memory = ConversationTokenBufferMemory(llm=self.llm, max_token_limit=3000)
|
25 |
|
26 |
# Load LLaMA-compatible tokenizer
|
27 |
+
tokenizer = AutoTokenizer.from_pretrained("gpt2")
|
28 |
|
29 |
# Define a simple token counting function
|
30 |
def llama_token_count(text: str) -> int:
|