more registry readme updates

This commit is contained in:
jeromeku 2025-03-31 18:34:18 -07:00
commit 521f50b854
2 changed files with 18 additions and 1 deletions

View file

@ -91,3 +91,20 @@ Prints the following (abridged) output:
...
```
### TODO
- Model Collections
- [x] Gemma3
- [ ] Llama3.1
- [x] Llama3.2
- [x] MistralSmall
- [x] Qwen2.5
- [x] Qwen2.5-VL
- [ ] Qwen2.5 Coder
- [x] QwenQwQ-32B
- [x] Deepseek v3
- [x] Deepseek R1
- [x] Phi-4
- [ ] Unsloth 4-bit Dynamic Quants
- [ ] Vision/multimodal models
- Sync model uploads with registry
- Add utility methods for tracking model stats

View file

@ -102,7 +102,7 @@ def register_llama_models(include_original_model: bool = False):
if __name__ == "__main__":
from unsloth.registry.registry import MODEL_REGISTRY, _check_model_info
MODEL_REGISTRY.clear()
register_llama_models(include_original_model=True)
for model_id, model_info in MODEL_REGISTRY.items():