tfrere commited on
Commit
87ff18b
·
1 Parent(s): 034bce7

use HF_ORGANIZATION env var instead of hardcoded 'yourbench' for billing

Browse files
backend/tasks/evaluation_task.py CHANGED
@@ -173,7 +173,7 @@ TASKS_TABLE = [yourbench]
173
  "lighteval",
174
  "endpoint",
175
  "inference-providers",
176
- f"model_name={model_name},provider={provider},org_to_bill=yourbench",
177
  "custom|yourbench|0|0",
178
  "--custom-tasks",
179
  temp_file_path,
 
173
  "lighteval",
174
  "endpoint",
175
  "inference-providers",
176
+ f"model_name={model_name},provider={provider},org_to_bill={os.getenv('HF_ORGANIZATION', 'yourbench')}",
177
  "custom|yourbench|0|0",
178
  "--custom-tasks",
179
  temp_file_path,
backend/tasks/get_available_model_provider.py CHANGED
@@ -128,7 +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
- bill_to="yourbench",
132
  timeout=3
133
  )
134
 
 
128
  client = InferenceClient(
129
  model=model_name,
130
  provider=provider,
131
+ bill_to=os.getenv('HF_ORGANIZATION', 'yourbench'),
132
  timeout=3
133
  )
134