Tencent · yghstill · Jun 1, 2026 · Mar 13, 2026 · Mar 13, 2026 · Mar 13, 2026
diff --git a/.gitignore b/.gitignore
@@ -15,10 +15,7 @@ eval/
 *_ckpt*/
 output/
 outputs/
-output*/
-logs*/
 outs/
 wandb/
 tools/results/
 __pycache__/outputs/
-__pycache__/
diff --git a/angelslim/compressor/speculative/train/data/data_utils.py b/angelslim/compressor/speculative/train/data/data_utils.py
@@ -91,7 +91,6 @@ def convert_ultrachat_data(row, dataset_column="messages"):
     return {"conversations": converted_messages, "id": row["prompt_id"]}
 
 
-# Copied from https://github.com/sgl-project/SpecForge/blob/main/specforge/data/preprocessing.py # noqa: E501
 def process_token_dict_to_mappings(
     token_dict,
     draft_vocab_size: int,

diff --git a/angelslim/compressor/speculative/train/models/draft/__init__.py b/angelslim/compressor/speculative/train/models/draft/__init__.py
@@ -14,6 +14,7 @@
 
 from .draft_model_factory import DraftModelConfig, create_draft_model
 from .llama_eagle3 import CosyVoice3Eagle3LlamaForCausalLM, Eagle3LlamaForCausalLM
+from .qwen_dflare import QwenDFlareDraftModel
 from .qwen_dflash import QwenDFlashDraftModel
 
 __all__ = [
@@ -22,4 +23,5 @@
     "Eagle3LlamaForCausalLM",
     "CosyVoice3Eagle3LlamaForCausalLM",
     "QwenDFlashDraftModel",
+    "QwenDFlareDraftModel",
 ]