Commit
·
33dfed7
1
Parent(s):
53154f9
changing to zephyr
Browse files
main.py
CHANGED
@@ -11,7 +11,7 @@ from fastapi import FastAPI
|
|
11 |
app = FastAPI()
|
12 |
|
13 |
# Define the inference model
|
14 |
-
llm = Anyscale(model="
|
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"))
|