Update src/rag_pipeline.py
Browse files- src/rag_pipeline.py +1 -1
src/rag_pipeline.py
CHANGED
|
@@ -29,7 +29,7 @@ class RAGPipeline:
|
|
| 29 |
|
| 30 |
# Initialize LLM with Gemma3Model
|
| 31 |
logger.debug("Loading Gemma 3 1B model...")
|
| 32 |
-
self.llm = Gemma3Model(model_name="
|
| 33 |
|
| 34 |
logger.info("✓ RAG Pipeline initialized successfully")
|
| 35 |
|
|
|
|
| 29 |
|
| 30 |
# Initialize LLM with Gemma3Model
|
| 31 |
logger.debug("Loading Gemma 3 1B model...")
|
| 32 |
+
self.llm = Gemma3Model(model_name="unsloth/gemma-3-1b-pt", device=device) # ← Use Gemma3Model
|
| 33 |
|
| 34 |
logger.info("✓ RAG Pipeline initialized successfully")
|
| 35 |
|