Leo Fan commited on
Commit
96ca499
1 Parent(s): 0504be5

Switched LLM to mistralai/Mistral-7B-Instruct-v0.2.

Browse files
Files changed (1) hide show
  1. 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/Mixtral-8x7B-v0.1",
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}