Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Alina Lozovskaya
commited on
Commit
·
be4c1de
1
Parent(s):
a22c06a
Set up bill_to hf_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 |
-
|
60 |
timeout=3 # Increased timeout to allow model loading
|
61 |
)
|
62 |
|
|
|
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 |
|