github/ColossalAI
1
0
Fork 0
mirror of https://github.com/hpcaitech/ColossalAI.git synced 2025-09-16 06:30:41 +00:00
Code Issues Packages Projects Releases Wiki Activity

[release] update version (#6195)

Browse Source 
* [release] update version

* fix test

* fix test
This commit is contained in:
Hongxin Liu
2025-02-20 11:36:18 +08:00
committed by GitHub
parent 24dee8f0b7
commit 9379cbd668
4 changed files with 14 additions and 25 deletions
Download Patch File Download Diff File Expand all files Collapse all files

12
tests/kit/model_zoo/transformers/deepseek_v3.py
View File

@@ -5,8 +5,6 @@ import torch
import transformers
from transformers import AutoConfig
from ..registry import ModelAttribute, model_zoo
# ===============================
# Register single-sentence Mixtral
# ===============================
@@ -75,13 +73,3 @@ def init_deepseek():
if m.__class__.__name__ == "DeepseekV3MoE":
m.moe_infer = MethodType(m.moe_infer.__wrapped__, m)
return model
model_zoo.register(
name="transformers_deepseek_v3",
model_fn=init_deepseek,
data_gen_fn=data_gen_for_lm,
output_transform_fn=output_transform_fn,
loss_fn=loss_fn_for_lm,
model_attribute=ModelAttribute(has_control_flow=True),
)