Search Hf Models
search_hf_modelsSearch Hugging Face Hub for open-source AI models by task or keyword and filter results by GPU VRAM limits and commercial-use licensing.
Instructions
Search for open-source AI models on Hugging Face Hub with hardware/VRAM requirements.
When to use:
Looking for ML/AI models for a specific task (e.g. 'automatic-speech-recognition', 'text-to-speech').
User has a local GPU limit (e.g. max_vram_gb=12.0) or needs commercial licensing.
Input Schema
| Name | Required | Description | Default |
|---|---|---|---|
| task | No | Optional pipeline_tag (e.g., 'automatic-speech-recognition', 'voice-activity-detection'). | |
| limit | No | Max candidates to return (default: 8). | |
| query | Yes | Model keyword (e.g., 'whisper', 'diarization', 'qwen'). | |
| max_vram_gb | No | Hardware ceiling in GB. Models exceeding this memory baseline are filtered out. | |
| commercial_use | No | True applies penalties for CC-BY-NC or custom restrictive licenses. |
Output Schema
| Name | Required | Description | Default |
|---|---|---|---|
| result | Yes |