Spaces:
Sleeping
Sleeping
Leo Fan
commited on
Commit
•
96ca499
1
Parent(s):
0504be5
Switched LLM to mistralai/Mistral-7B-Instruct-v0.2.
Browse files- app/server.py +1 -1
app/server.py
CHANGED
@@ -14,7 +14,7 @@ app = FastAPI()
|
|
14 |
# os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
|
15 |
|
16 |
hf_llm = HuggingFaceHub(
|
17 |
-
repo_id="mistralai/
|
18 |
huggingfacehub_api_token=os.environ["HF_TOKEN"],
|
19 |
task="text-generation",
|
20 |
model_kwargs={"temperature":0.01, "max_new_tokens" : 250}
|
|
|
14 |
# os.environ['TRANSFORMERS_CACHE'] = '/blabla/cache/'
|
15 |
|
16 |
hf_llm = HuggingFaceHub(
|
17 |
+
repo_id="mistralai/Mistral-7B-Instruct-v0.2",
|
18 |
huggingfacehub_api_token=os.environ["HF_TOKEN"],
|
19 |
task="text-generation",
|
20 |
model_kwargs={"temperature":0.01, "max_new_tokens" : 250}
|