Spaces:
Paused
Paused
change model
Browse files- rag_pipeline.py +1 -1
rag_pipeline.py
CHANGED
@@ -23,7 +23,7 @@ def initialize_components(data_path):
|
|
23 |
# 1. Tải LLM và Tokenizer từ Unsloth
|
24 |
print("1. Tải mô hình LLM (Unsloth)...")
|
25 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
26 |
-
model_name="unsloth/Llama-
|
27 |
max_seq_length=2048,
|
28 |
dtype=None,
|
29 |
load_in_4bit=True,
|
|
|
23 |
# 1. Tải LLM và Tokenizer từ Unsloth
|
24 |
print("1. Tải mô hình LLM (Unsloth)...")
|
25 |
model, tokenizer = FastLanguageModel.from_pretrained(
|
26 |
+
model_name="unsloth/Llama-3.2-11B-Vision-Instruct-unsloth-bnb-4bit",
|
27 |
max_seq_length=2048,
|
28 |
dtype=None,
|
29 |
load_in_4bit=True,
|