diff --git a/unsloth/registry/REGISTRY.md b/unsloth/registry/REGISTRY.md index 8240d686e6..a0b3d96cad 100644 --- a/unsloth/registry/REGISTRY.md +++ b/unsloth/registry/REGISTRY.md @@ -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 \ No newline at end of file diff --git a/unsloth/registry/_llama.py b/unsloth/registry/_llama.py index 1c2dd5bf18..f1b9dbdd32 100644 --- a/unsloth/registry/_llama.py +++ b/unsloth/registry/_llama.py @@ -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():