datacipen commited on
Commit
64d0204
1 Parent(s): 04d41c9

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -49,7 +49,8 @@ def auth_callback(username: str, password: str):
49
  @cl.step(type="tool")
50
  async def LLModel():
51
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
52
- repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
 
53
  llm = HuggingFaceEndpoint(
54
  repo_id=repo_id, max_new_tokens=5300, temperature=0.5, task="text2text-generation", streaming=True
55
  )
 
49
  @cl.step(type="tool")
50
  async def LLModel():
51
  os.environ['HUGGINGFACEHUB_API_TOKEN'] = os.environ['HUGGINGFACEHUB_API_TOKEN']
52
+ #repo_id = "mistralai/Mixtral-8x7B-Instruct-v0.1"
53
+ repo_id = "mistralai/Mistral-7B-Instruct-v0.3"
54
  llm = HuggingFaceEndpoint(
55
  repo_id=repo_id, max_new_tokens=5300, temperature=0.5, task="text2text-generation", streaming=True
56
  )