KJ24 commited on
Commit
a66e2c5
·
verified ·
1 Parent(s): 51c0208

Update chunker_pipeline.py

Browse files
Files changed (1) hide show
  1. chunker_pipeline.py +7 -4
chunker_pipeline.py CHANGED
@@ -40,6 +40,7 @@ from llama_index.embeddings.huggingface import HuggingFaceEmbedding
40
  from llama_index.core.settings import Settings
41
 
42
 
 
43
  from llama_index.llms.llama_cpp import LlamaCPP # ✅ CORRECT
44
 
45
 
@@ -450,12 +451,14 @@ class SmartChunkerPipeline:
450
  # Settings.embed_model = self.embed_model
451
 
452
 
453
- # self.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
454
- # Settings.embed_model = self.embed_model
455
-
456
- self.embed_model = EmbeddingWrapper(HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2"))
457
  Settings.embed_model = self.embed_model
458
 
 
 
 
 
 
459
  # 🔐 Test de sécurité complet
460
 
461
  # try:
 
40
  from llama_index.core.settings import Settings
41
 
42
 
43
+
44
  from llama_index.llms.llama_cpp import LlamaCPP # ✅ CORRECT
45
 
46
 
 
451
  # Settings.embed_model = self.embed_model
452
 
453
 
454
+ self.embed_model = HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2")
 
 
 
455
  Settings.embed_model = self.embed_model
456
 
457
+ # self.embed_model = EmbeddingWrapper(HuggingFaceEmbedding(model_name="sentence-transformers/all-MiniLM-L6-v2"))
458
+ # Settings.embed_model = self.embed_model
459
+
460
+
461
+
462
  # 🔐 Test de sécurité complet
463
 
464
  # try: