Update all_models2.py

all_models2.py

@@ -33,6 +33,32 @@ def find_model_list(author: str="", tags: list[str]=[], not_tag="", sort: str="l
             if len(models) == limit: break
     return models , models_plus_tags
 
+def find_warm_model_list(author: str="", tags: list[str]=[], not_tag="", sort: str="last_modified", limit: int=30, force_gpu=False, check_status=False, bad_models=bad_models):
+    from huggingface_hub import HfApi
+    api = HfApi()
+    default_tags = ["diffusers"]
+    if not sort: sort = "last_modified"
+    models = []
+    models_plus_tags=[]
+    try:
+        model_infos = api.list_models(author=author, task="text-to-image",
+                                       tags=list_uniq(default_tags + tags), cardData=True, sort=sort, limit=limit, inference ="warm")
+    except Exception as e:
+        print(f"Error: Failed to list models.")
+        print(e)
+        return models
+    for model in model_infos:
+        
+        if not model.private and not model.gated:
+            loadable = True
+            if not_tag and not_tag in model.tags or not loadable: continue
+            if model.id not in bad_models :
+                models.append(model.id)
+                #models_plus_tags.append([model.id,process_tags(model.tags)])
+                models_plus_tags.append([model.id,process_tags(model.cardData.tags)])
+            if len(models) == limit: break
+    return models , models_plus_tags
+
 def process_tags(tags,bad_tags=bad_tags):
     t1=True
     new_tags=[]