Spaces:
Sleeping
Sleeping
Update src/RAGSample.py
Browse files- src/RAGSample.py +2 -2
src/RAGSample.py
CHANGED
@@ -411,9 +411,9 @@ def setup_rag_chain() -> Runnable:
|
|
411 |
|
412 |
|
413 |
try:
|
414 |
-
tokenizer = AutoTokenizer.from_pretrained("
|
415 |
model = AutoModelForCausalLM.from_pretrained(
|
416 |
-
"
|
417 |
device_map="auto",
|
418 |
torch_dtype=torch.float16
|
419 |
)
|
|
|
411 |
|
412 |
|
413 |
try:
|
414 |
+
tokenizer = AutoTokenizer.from_pretrained("HuggingFaceTB/SmolLM3-3B")
|
415 |
model = AutoModelForCausalLM.from_pretrained(
|
416 |
+
"HuggingFaceTB/SmolLM3-3B",
|
417 |
device_map="auto",
|
418 |
torch_dtype=torch.float16
|
419 |
)
|