Sadiaa commited on
Commit
5c31faf
·
verified ·
1 Parent(s): 9c9a42b

Update chatbot.py

Browse files
Files changed (1) hide show
  1. 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("meta-llama/Llama-2-7b-chat-hf")
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: