Alina Lozowski commited on
Commit
e7893a8
1 Parent(s): 5a5efd9

Rename curated authors to official providers

Browse files
Files changed (1) hide show
  1. src/submission/check_validity.py +3 -3
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, curated_authors
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 curated authors
131
- if org_or_user in curated_authors:
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: