Spaces:
Running
on
CPU Upgrade
Running
on
CPU Upgrade
Alina Lozowski
commited on
Commit
·
e7893a8
1
Parent(s):
5a5efd9
Rename curated authors to official providers
Browse files
src/submission/check_validity.py
CHANGED
|
@@ -10,7 +10,7 @@ from huggingface_hub import ModelCard, hf_hub_download
|
|
| 10 |
from huggingface_hub.hf_api import ModelInfo, get_safetensors_metadata, parse_safetensors_file_metadata
|
| 11 |
from transformers import AutoConfig, AutoTokenizer
|
| 12 |
|
| 13 |
-
from src.display.utils import parse_iso8601_datetime,
|
| 14 |
from src.envs import HAS_HIGHER_RATE_LIMIT
|
| 15 |
|
| 16 |
|
|
@@ -127,8 +127,8 @@ def get_model_arch(model_info: ModelInfo):
|
|
| 127 |
|
| 128 |
|
| 129 |
def user_submission_permission(org_or_user, users_to_submission_dates, rate_limit_period, rate_limit_quota):
|
| 130 |
-
# No limit for
|
| 131 |
-
if org_or_user in
|
| 132 |
return True, ""
|
| 133 |
# Increase quota first if user has higher limits
|
| 134 |
if org_or_user in HAS_HIGHER_RATE_LIMIT:
|
|
|
|
| 10 |
from huggingface_hub.hf_api import ModelInfo, get_safetensors_metadata, parse_safetensors_file_metadata
|
| 11 |
from transformers import AutoConfig, AutoTokenizer
|
| 12 |
|
| 13 |
+
from src.display.utils import parse_iso8601_datetime, official_providers
|
| 14 |
from src.envs import HAS_HIGHER_RATE_LIMIT
|
| 15 |
|
| 16 |
|
|
|
|
| 127 |
|
| 128 |
|
| 129 |
def user_submission_permission(org_or_user, users_to_submission_dates, rate_limit_period, rate_limit_quota):
|
| 130 |
+
# No limit for official providers
|
| 131 |
+
if org_or_user in official_providers:
|
| 132 |
return True, ""
|
| 133 |
# Increase quota first if user has higher limits
|
| 134 |
if org_or_user in HAS_HIGHER_RATE_LIMIT:
|