Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
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=
|
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 |
|