alexkueck commited on
Commit
f489ec0
1 Parent(s): c7383c7

Update utils.py

Browse files
Files changed (1) hide show
  1. utils.py +2 -2
utils.py CHANGED
@@ -166,8 +166,8 @@ urls = [
166
  ##################################################
167
  # Schritt 1: Embedding Modelle, um relvante Texte zu einem Prompt zu finden
168
  #sowohl die texte als auch der Prompt werden embeddet!
169
- embedder_modell = SentenceTransformer("sentence-transformers/all-mpnet-base-v2") #all-MiniLM-L6-v2
170
- EMBEDDING_MODELL = "sentence-transformers/all-mpnet-base-v2"
171
 
172
  #Modell und Tokenizer, um die Summary über die relevanten Texte zu machen
173
  #mögliche Modelle: "HuggingFaceH4/zephyr-7b-alpha" #"t5-small" #"meta-llama/Meta-Llama-3-8B-Instruct" #"mistralai/Mistral-7B-Instruct-v0.3" #"microsoft/Phi-3-mini-4k-instruct" #"HuggingFaceH4/zephyr-7b-alpha"
 
166
  ##################################################
167
  # Schritt 1: Embedding Modelle, um relvante Texte zu einem Prompt zu finden
168
  #sowohl die texte als auch der Prompt werden embeddet!
169
+ embedder_modell = SentenceTransformer("multi-qa-MPNet-base-dot-v1") #all-MiniLM-L6-v2 sentence-transformers/all-mpnet-base-v2
170
+ EMBEDDING_MODELL = "multi-qa-MPNet-base-dot-v1"
171
 
172
  #Modell und Tokenizer, um die Summary über die relevanten Texte zu machen
173
  #mögliche Modelle: "HuggingFaceH4/zephyr-7b-alpha" #"t5-small" #"meta-llama/Meta-Llama-3-8B-Instruct" #"mistralai/Mistral-7B-Instruct-v0.3" #"microsoft/Phi-3-mini-4k-instruct" #"HuggingFaceH4/zephyr-7b-alpha"