diff --git a/README.md b/README.md index 44d072e..b24b38e 100644 --- a/README.md +++ b/README.md @@ -241,8 +241,3 @@ peft_model = PeftModel.from_pretrained(model, 'Qwen3.5-4B-LoRA') ## 🏛 License This framework is licensed under the [Apache License (Version 2.0)](https://github.com/modelscope/mcore-bridge/blob/master/LICENSE). For models and datasets, please refer to the original resource page and follow the corresponding License. - - -## Star History - -[![Star History Chart](https://api.star-history.com/svg?repos=modelscope/mcore-bridge&type=Date)](https://star-history.com/#modelscope/mcore-bridge&Date) diff --git a/README_zh.md b/README_zh.md index 801b4af..2f1af5a 100644 --- a/README_zh.md +++ b/README_zh.md @@ -238,8 +238,3 @@ peft_model = PeftModel.from_pretrained(model, 'Qwen3.5-4B-LoRA') ## 🏛 License 本框架使用[Apache License (Version 2.0)](https://github.com/modelscope/mcore-bridge/blob/master/LICENSE)进行许可。模型和数据集请查看原资源页面并遵守对应License。 - - -## Star History - -[![Star History Chart](https://api.star-history.com/svg?repos=modelscope/mcore-bridge&type=Date)](https://star-history.com/#modelscope/mcore-bridge&Date) diff --git a/src/mcore_bridge/patcher.py b/src/mcore_bridge/patcher.py index f2009a1..15280b5 100644 --- a/src/mcore_bridge/patcher.py +++ b/src/mcore_bridge/patcher.py @@ -689,8 +689,7 @@ def forward_with_scores( try: from megatron.core.transformer.experimental_attention_variant.dsa import fused_qk_topk_naive except ImportError: - raise ImportError('fused_qk_topk_naive is not available. Please install megatron-core from source. ' - '`pip install git+https://github.com/NVIDIA/Megatron-LM.git`') + raise ImportError('fused_qk_topk_naive is not available. Please install "megatron-core>=0.17.0"') # [seqlen, batch, index_n_heads * index_head_dim] # [seqlen, batch, index_head_dim] # [seqlen, batch, index_n_heads]