Upload 2 files

modutils.py

@@ -508,12 +508,18 @@ def get_t2i_model_info(repo_id: str):
     return gr.update(value=md)
 
 
+MAX_MODEL_INFO = 100
+
+
 def get_tupled_model_list(model_list):
     if not model_list: return []
     #return [(x, x) for x in model_list] # for skipping this function
     tupled_list = []
-    for repo_id in model_list:
-        api = HfApi()
+    api = HfApi()
+    for i, repo_id in enumerate(model_list):
+        if i > MAX_MODEL_INFO:
+            tupled_list.append((repo_id, repo_id))
+            continue
         try:
             if not api.repo_exists(repo_id): continue
             model = api.model_info(repo_id=repo_id, timeout=0.5)
@@ -521,7 +527,7 @@ def get_tupled_model_list(model_list):
             print(f"{repo_id}: {e}")
             tupled_list.append((repo_id, repo_id))
             continue
-        if model.private or model.gated: continue
+        if model.tags is None: continue
         tags = model.tags
         info = []
         if not 'diffusers' in tags: continue

requirements.txt

@@ -1,5 +1,5 @@
 stablepy==0.6.1
-diffusers>=0.31.0
+diffusers<=0.32.0
 transformers==4.47.1
 torch==2.4.0
 numpy<2
@@ -10,6 +10,8 @@ accelerate
 optimum[onnxruntime]
 dartrs
 huggingface_hub
+hf_transfer
+hf_xet
 translatepy
 timm
 rapidfuzz