Waflon commited on
Commit
44337bd
1 Parent(s): 2634869

Update modelo.py

Browse files
Files changed (1) hide show
  1. modelo.py +1 -1
modelo.py CHANGED
@@ -12,7 +12,7 @@ def get_chain():
12
  # agregada en la config de hugginface
13
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
14
  embeddings = HuggingFaceEmbeddings(
15
- model_name="sentence-transformers/multi-qa-mpnet-base-dot-v1", # Modelo correcto para embeddings que serán usados como QA
16
  #model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
17
  model_kwargs={'device':'cpu'}, # Opciones de configuracion del modelo
18
  encode_kwargs={'normalize_embeddings': False} # Opciones de Encoding
 
12
  # agregada en la config de hugginface
13
  #Embeddings que transforman a vectores densos multidimensionales las preguntas del SII
14
  embeddings = HuggingFaceEmbeddings(
15
+ model_name="sentence-transformers/multi-qa-MiniLM-L6-cos-v1", # Modelo correcto para embeddings que serán usados como QA
16
  #model_name="sentence-transformers/paraphrase-multilingual-MiniLM-L12-v2", # Ruta a modelo Pre entrenado
17
  model_kwargs={'device':'cpu'}, # Opciones de configuracion del modelo
18
  encode_kwargs={'normalize_embeddings': False} # Opciones de Encoding