rishikumar20202023 commited on
Commit
3a40ffd
·
verified ·
1 Parent(s): b07c745

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -78,8 +78,8 @@ subprocess.run(
78
 
79
 
80
  # Load the tokenizer and model
81
- tokenizer = AutoTokenizer.from_pretrained("HumanLLMs/Human-Like-LLama3-8B-Instruct")
82
- model = AutoModelForCausalLM.from_pretrained("HumanLLMs/Human-Like-LLama3-8B-Instruct", device_map="auto",
83
  low_cpu_mem_usage=True,
84
  torch_dtype=torch.bfloat16,
85
  # quantization_config=quantization_config,
 
78
 
79
 
80
  # Load the tokenizer and model
81
+ tokenizer = AutoTokenizer.from_pretrained("HumanLLMs/Human-Like-Mistral-Nemo-Instruct-2407")
82
+ model = AutoModelForCausalLM.from_pretrained("HumanLLMs/Human-Like-Mistral-Nemo-Instruct-2407", device_map="auto",
83
  low_cpu_mem_usage=True,
84
  torch_dtype=torch.bfloat16,
85
  # quantization_config=quantization_config,