File size: 1,078 Bytes
51545af 269ea1f 51545af 269ea1f 51545af | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 | def list_available_models():
"""List all available models."""
# Create list of all models
models_data = []
# Add special models first (Imageomics BioCLIP family)
models_data.extend([
{"name": "hf-hub:imageomics/bioclip-2", "pretrained": None},
{"name": "hf-hub:imageomics/bioclip-2.5-vith14", "pretrained": None},
{"name": "hf-hub:imageomics/biocap", "pretrained": None},
{"name": "hf-hub:imageomics/bioclip", "pretrained": None}
])
# OpenCLIP models
import open_clip
openclip_models = open_clip.list_pretrained()
for model_name, pretrained in openclip_models:
models_data.append({
"name": model_name,
"pretrained": pretrained
})
return models_data
def print_available_models():
"""CLI entry point: print all available models to stdout."""
models = list_available_models()
for m in models:
if m["pretrained"]:
print(f"{m['name']} (pretrained: {m['pretrained']})")
else:
print(m["name"])
|