brendon-ai commited on
Commit
56328b4
·
verified ·
1 Parent(s): 9fb2feb

Update src/RAGSample.py

Browse files
Files changed (1) hide show
  1. src/RAGSample.py +2 -2
src/RAGSample.py CHANGED
@@ -371,9 +371,9 @@ Answer:
371
  )
372
 
373
  try:
374
- tokenizer = AutoTokenizer.from_pretrained("microsoft/BioGPT")
375
  model = AutoModelForCausalLM.from_pretrained(
376
- "microsoft/BioGPT",
377
  device_map="auto",
378
  torch_dtype=torch.float16
379
  )
 
371
  )
372
 
373
  try:
374
+ tokenizer = AutoTokenizer.from_pretrained("meta-llama/Llama-3.2-3B-Instruct")
375
  model = AutoModelForCausalLM.from_pretrained(
376
+ "meta-llama/Llama-3.2-3B-Instruct",
377
  device_map="auto",
378
  torch_dtype=torch.float16
379
  )