lixuejing commited on
Commit
70fec46
1 Parent(s): 7f7c084

fix ModelFilter bug

Browse files
requirements.txt CHANGED
@@ -5,7 +5,7 @@ datasets==2.14.5
5
  gradio==4.9.0
6
  gradio_client==0.7.2
7
  TensorFlow >= 2.0
8
- huggingface-hub==0.23.5
9
  matplotlib==3.7.1
10
  numpy==1.24.2
11
  pandas==2.0.0
 
5
  gradio==4.9.0
6
  gradio_client==0.7.2
7
  TensorFlow >= 2.0
8
+ huggingface-hub>=0.25.0
9
  matplotlib==3.7.1
10
  numpy==1.24.2
11
  pandas==2.0.0
src/scripts/update_all_request_files.py CHANGED
@@ -1,4 +1,5 @@
1
- from huggingface_hub import ModelFilter, snapshot_download
 
2
  from huggingface_hub import ModelCard
3
 
4
  import json
@@ -66,7 +67,8 @@ def update_dynamic_files():
66
  start = time.time()
67
 
68
  models = list(API.list_models(
69
- filter=ModelFilter(task="text-generation"),
 
70
  full=False,
71
  cardData=True,
72
  fetch_config=True,
 
1
+ #from huggingface_hub import ModelFilter, snapshot_download
2
+ from huggingface_hub import snapshot_download
3
  from huggingface_hub import ModelCard
4
 
5
  import json
 
67
  start = time.time()
68
 
69
  models = list(API.list_models(
70
+ #filter=ModelFilter(task="text-generation"),
71
+ task="text-generation",
72
  full=False,
73
  cardData=True,
74
  fetch_config=True,