datacipen commited on
Commit
3c312a7
·
verified ·
1 Parent(s): 4464705

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -121,7 +121,8 @@ def API_France_Travail_Metier(metier):
121
  def Connexion_Mistral():
122
  #endpoint = "https://models.inference.ai.azure.com"
123
  #return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
124
- repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
 
125
  #repo_id = "Qwen/Qwen2.5-3B-Instruct"
126
  #repo_id = "HuggingFaceH4/zephyr-7b-beta"
127
  #repo_id = "HuggingFaceH4/zephyr-7b-alpha"
 
121
  def Connexion_Mistral():
122
  #endpoint = "https://models.inference.ai.azure.com"
123
  #return Mistral(api_key=os.environ["GITHUB_TOKEN"], server_url=endpoint)
124
+ #repo_id = "mistralai/Mistral-7B-Instruct-v0.2"
125
+ repo_id = "NousResearch/Yarn-Mistral-7b-128k"
126
  #repo_id = "Qwen/Qwen2.5-3B-Instruct"
127
  #repo_id = "HuggingFaceH4/zephyr-7b-beta"
128
  #repo_id = "HuggingFaceH4/zephyr-7b-alpha"