sumuks commited on
Commit
8d5d6ef
·
1 Parent(s): be4c1de

hardcode organization

Browse files
backend/tasks/get_available_model_provider.py CHANGED
@@ -56,7 +56,7 @@ def test_provider(model_name: str, provider: str, verbose: bool = False) -> bool
56
  model=model_name,
57
  token=hf_token,
58
  provider=provider,
59
- bill_to=hf_organization if hf_organization else None,
60
  timeout=3 # Increased timeout to allow model loading
61
  )
62
 
@@ -128,6 +128,7 @@ def _test_provider_without_token(model_name: str, provider: str, verbose: bool =
128
  client = InferenceClient(
129
  model=model_name,
130
  provider=provider,
 
131
  timeout=3
132
  )
133
 
 
56
  model=model_name,
57
  token=hf_token,
58
  provider=provider,
59
+ bill_to=hf_organization if hf_organization else "yourbench",
60
  timeout=3 # Increased timeout to allow model loading
61
  )
62
 
 
128
  client = InferenceClient(
129
  model=model_name,
130
  provider=provider,
131
+ bill_to="yourbench",
132
  timeout=3
133
  )
134