pbrenotvinciguerra commited on
Commit
c4b67c7
1 Parent(s): 4f265da

back to mistral 7b model

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -10,8 +10,8 @@ from llama_index.query_engine import RetrieverQueryEngine
10
  from llama_index.llms import Anyscale
11
 
12
  # Define the inference model
13
- # llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", api_key=os.getenv("ANYSCALE_API_KEY"))
14
- llm = Anyscale(model="HuggingFaceH4/zephyr-7b-beta", api_key=os.getenv("ANYSCALE_API_KEY"))
15
  # Define the embedding model used to embed the query.
16
  # query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
17
  embed_model = VoyageEmbedding(model_name="voyage-01", voyage_api_key=os.getenv("VOYAGE_API_KEY"))
 
10
  from llama_index.llms import Anyscale
11
 
12
  # Define the inference model
13
+ llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", api_key=os.getenv("ANYSCALE_API_KEY"))
14
+ #llm = Anyscale(model="HuggingFaceH4/zephyr-7b-beta", api_key=os.getenv("ANYSCALE_API_KEY"))
15
  # Define the embedding model used to embed the query.
16
  # query_embed_model = HuggingFaceEmbedding(model_name="BAAI/bge-base-en-v1.5")
17
  embed_model = VoyageEmbedding(model_name="voyage-01", voyage_api_key=os.getenv("VOYAGE_API_KEY"))