diff --git a/privateGPT.py b/privateGPT.py index 7d97a56..5440ce8 100644 --- a/privateGPT.py +++ b/privateGPT.py @@ -9,7 +9,7 @@ import argparse import time -model = os.environ.get("MODEL", "llama2-uncensored") +model = os.environ.get("MODEL", "llama2") # For embeddings model, the example uses a sentence-transformers model # https://www.sbert.net/docs/pretrained_models.html # "The all-mpnet-base-v2 model provides the best quality, while all-MiniLM-L6-v2 is 5 times faster and still offers good quality."