dnj0 commited on
Commit
4396f57
·
verified ·
1 Parent(s): bb6a42c

Update src/rag_pipeline.py

Browse files
Files changed (1) hide show
  1. 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="google/gemma-3-1b-it", device=device) # ← Use Gemma3Model
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