From 3f8cefca1a73d53d9cdfa5ed5a0b5525077bc118 Mon Sep 17 00:00:00 2001 From: Taksh Date: Wed, 24 Jun 2026 17:08:27 +0530 Subject: [PATCH] fix(taggers): use HTTPS URL for code composition FastText model smart_open cannot fetch hf:// URIs, so CodeProseCompositionClassifier failed to load its model at instantiation time. Co-authored-by: Cursor --- python/dolma/taggers/code_composition.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/dolma/taggers/code_composition.py b/python/dolma/taggers/code_composition.py index c8198b41..f1782934 100644 --- a/python/dolma/taggers/code_composition.py +++ b/python/dolma/taggers/code_composition.py @@ -24,7 +24,7 @@ @TaggerRegistry.add("code_composition") class CodeProseCompositionClassifier(BaseFastTextTagger): - MODEL_PATH = "hf://techarb/code-prose-composition/code-comment-prose-model.bin" # noqa: E501 + MODEL_PATH = "https://huggingface.co/techarb/code-prose-composition/resolve/main/code-comment-prose-model.bin" # noqa: E501 def __init__(self): super().__init__(model_path=self.MODEL_PATH, model_mode=self.DOCUMENT_LEVEL_TAGGER)