rishikumar20202023 commited on
Commit
f897284
·
verified ·
1 Parent(s): b79c4e7

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -61,8 +61,8 @@ def str_to_json(str_obj):
61
 
62
 
63
  # Load the tokenizer and model
64
- tokenizer = AutoTokenizer.from_pretrained("HumanLLMs/Human-Like-Qwen2.5-7B-Instruct")
65
- model = AutoModelForCausalLM.from_pretrained("HumanLLMs/Human-Like-Qwen2.5-7B-Instruct", device_map="auto") # to("cuda:0")
66
  terminators = [
67
  tokenizer.eos_token_id,
68
  tokenizer.convert_tokens_to_ids("<|eot_id|>")
 
61
 
62
 
63
  # Load the tokenizer and model
64
+ tokenizer = AutoTokenizer.from_pretrained("HumanLLMs/Human-Like-Mistral-Nemo-Instruct-2407")
65
+ model = AutoModelForCausalLM.from_pretrained("HumanLLMs/Human-Like-Mistral-Nemo-Instruct-2407", device_map="auto") # to("cuda:0")
66
  terminators = [
67
  tokenizer.eos_token_id,
68
  tokenizer.convert_tokens_to_ids("<|eot_id|>")