if001
/

DeepSeek-R1-0528-tiny

Text Generation

text-generation-inference

Model card Files Files and versions

if001 commited on Aug 21

Commit

b247db3

·

1 Parent(s): daf451b

fix method

Files changed (1) hide show

modeling_deepseek.py +1 -1

modeling_deepseek.py CHANGED Viewed

@@ -399,7 +399,7 @@ class MoEGate(nn.Module):
         self.routed_scaling_factor = config.routed_scaling_factor
         self.scoring_func = config.scoring_func
         # self.topk_method = config.topk_method
-        self.topk_method == "gready"
         self.n_group = config.n_group
         self.topk_group = config.topk_group

         self.routed_scaling_factor = config.routed_scaling_factor
         self.scoring_func = config.scoring_func
         # self.topk_method = config.topk_method
+        self.topk_method = "gready"
         self.n_group = config.n_group
         self.topk_group = config.topk_group