pbrenotvinciguerra commited on
Commit
33dfed7
·
1 Parent(s): 53154f9

changing to zephyr

Browse files
Files changed (1) hide show
  1. main.py +1 -1
main.py CHANGED
@@ -11,7 +11,7 @@ from fastapi import FastAPI
11
  app = FastAPI()
12
 
13
  # Define the inference model
14
- llm = Anyscale(model="mistralai/Mistral-7B-Instruct-v0.1", 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"))
 
11
  app = FastAPI()
12
 
13
  # Define the inference model
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"))