Clémentine commited on
Commit
f3cda22
1 Parent(s): f69c85c

token for checking gated base models

Browse files
Files changed (2) hide show
  1. app.py +1 -1
  2. src/filters.py +3 -2
app.py CHANGED
@@ -140,7 +140,7 @@ def add_new_eval(
140
  revision = "main"
141
 
142
  if weight_type in ["Delta", "Adapter"]:
143
- base_model_on_hub, error = is_model_on_hub(base_model, revision)
144
  if not base_model_on_hub:
145
  return styled_error(f'Base model "{base_model}" {error}')
146
 
 
140
  revision = "main"
141
 
142
  if weight_type in ["Delta", "Adapter"]:
143
+ base_model_on_hub, error = is_model_on_hub(base_model, revision, H4_TOKEN)
144
  if not base_model_on_hub:
145
  return styled_error(f'Base model "{base_model}" {error}')
146
 
src/filters.py CHANGED
@@ -1,4 +1,5 @@
1
  import huggingface_hub
 
2
  from huggingface_hub import ModelCard
3
  from transformers import AutoConfig
4
 
@@ -29,9 +30,9 @@ def check_model_card(repo_id: str) -> tuple[bool, str]:
29
  return True, ""
30
 
31
 
32
- def is_model_on_hub(model_name: str, revision: str) -> bool:
33
  try:
34
- AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=False)
35
  return True, None
36
 
37
  except ValueError:
 
1
  import huggingface_hub
2
+ import os
3
  from huggingface_hub import ModelCard
4
  from transformers import AutoConfig
5
 
 
30
  return True, ""
31
 
32
 
33
+ def is_model_on_hub(model_name: str, revision: str, token: str = None) -> bool:
34
  try:
35
+ AutoConfig.from_pretrained(model_name, revision=revision, trust_remote_code=False, token=token)
36
  return True, None
37
 
38
  except ValueError: