harsh-manvar commited on
Commit
3da0af1
1 Parent(s): d40f94b

Update model.py

Browse files
Files changed (1) hide show
  1. model.py +1 -2
model.py CHANGED
@@ -16,8 +16,7 @@ device = "cpu"
16
 
17
 
18
  model = AutoModelForCausalLM.from_pretrained(model_id, model_type="llama", lib='avx2', hf=True)
19
- tokenizer = AutoTokenizer.from_pretrained('Llama-2-7b-chat-hf')
20
- #tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
21
 
22
  def get_prompt(message: str, chat_history: list[tuple[str, str]],
23
  system_prompt: str) -> str:
 
16
 
17
 
18
  model = AutoModelForCausalLM.from_pretrained(model_id, model_type="llama", lib='avx2', hf=True)
19
+ tokenizer = AutoTokenizer.from_pretrained('meta-llama/Llama-2-7b-chat-hf')
 
20
 
21
  def get_prompt(message: str, chat_history: list[tuple[str, str]],
22
  system_prompt: str) -> str: