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