Ability to list models

rshf/__init__.py

@@ -0,0 +1 @@
+from .utils import list_models

rshf/senclip/model.py

@@ -23,8 +23,4 @@ def __init__(self, config: PretrainedConfig):
             self.config = PretrainedConfig().from_dict(config)
 
         self.device = "cpu" if not torch.cuda.is_available() else self.config.device
-        self.clip, _ = clip.load(self.config.architecture, device=self.device)
-
-
-
-
+        self.clip, _ = clip.load(self.config.architecture, device=self.device)

rshf/utils.py

@@ -0,0 +1,13 @@
+from huggingface_hub import HfApi
+import ssl
+
+#HACK: Ignore SSL errors
+ssl._create_default_https_context = ssl._create_unverified_context
+
+
+def list_models(model_name):
+    model_filter = lambda model_info: "mvrl" in model_info.modelId.lower() and model_name.lower() in model_info.modelId.lower()
+    models = filter(model_filter, HfApi().list_models())
+    print(f"Available {model_name} pretrained models:")
+    for model_info in models:
+        print(model_info.modelId)

setup.cfg

@@ -1,6 +1,6 @@
 [metadata]
 name = rshf
-version = 0.1.5
+version = 0.1.6
 author = Srikumar Sastry
 author_email = s.sastry@wustl.edu
 description = RS pretrained models in huggingface style