Spaces:
Sleeping
Sleeping
Update src/RAGSample.py
Browse files- src/RAGSample.py +2 -2
src/RAGSample.py
CHANGED
@@ -371,9 +371,9 @@ Answer:
|
|
371 |
)
|
372 |
|
373 |
try:
|
374 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
375 |
model = AutoModelForCausalLM.from_pretrained(
|
376 |
-
"
|
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 |
)
|