Daniel Nichols commited on
Commit
903c035
1 Parent(s): 972361d

remove gemma from list of models

Browse files
Files changed (1) hide show
  1. src/models.py +1 -1
src/models.py CHANGED
@@ -164,7 +164,7 @@ if os.environ.get("GOOGLE_API_KEY"):
164
 
165
  if os.environ.get("HF_API_KEY"):
166
  hf_inference_client = InferenceClient(api_key=os.environ.get("HF_API_KEY"), timeout=60)
167
- AVAILABLE_MODELS.append( InferenceHubModel("google/gemma-2-2b-it", hf_inference_client, supports_system_messages=False) )
168
  #AVAILABLE_MODELS.append( InferenceHubModel("Qwen/Qwen2.5-7B-Instruct", hf_inference_client) )
169
  AVAILABLE_MODELS.append( InferenceHubModel("microsoft/Phi-3-mini-4k-instruct", hf_inference_client) )
170
  AVAILABLE_MODELS.append( InferenceHubModel("meta-llama/Llama-3.2-1B-Instruct", hf_inference_client) )
 
164
 
165
  if os.environ.get("HF_API_KEY"):
166
  hf_inference_client = InferenceClient(api_key=os.environ.get("HF_API_KEY"), timeout=60)
167
+ #AVAILABLE_MODELS.append( InferenceHubModel("google/gemma-2-2b-it", hf_inference_client, supports_system_messages=False) )
168
  #AVAILABLE_MODELS.append( InferenceHubModel("Qwen/Qwen2.5-7B-Instruct", hf_inference_client) )
169
  AVAILABLE_MODELS.append( InferenceHubModel("microsoft/Phi-3-mini-4k-instruct", hf_inference_client) )
170
  AVAILABLE_MODELS.append( InferenceHubModel("meta-llama/Llama-3.2-1B-Instruct", hf_inference_client) )