datacipen commited on
Commit
38af760
·
verified ·
1 Parent(s): e8e9eec

Update main.py

Browse files
Files changed (1) hide show
  1. main.py +2 -1
main.py CHANGED
@@ -122,7 +122,8 @@ 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
  llm = HuggingFaceEndpoint(
127
  repo_id=repo_id, max_new_tokens=5000, temperature=0.1, task="text2text-generation", streaming=True
128
  )
 
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 = "microsoft/Phi-3.5-mini-instruct"
127
  llm = HuggingFaceEndpoint(
128
  repo_id=repo_id, max_new_tokens=5000, temperature=0.1, task="text2text-generation", streaming=True
129
  )