Update modeling_llada2_moe.py

modeling_llada2_moe.py

@@ -1239,11 +1239,11 @@ class LLaDA2MoeModelLM(LLaDA2MoePreTrainedModel, GenerationMixin):
     def generate(
         self,
         inputs: Optional[torch.Tensor] = None,
-        temperature: int = 0.0,
+        temperature: float = 0.0,
         block_length: int = 32,
         steps: int = 32,
         gen_length: int = 2048,
-        top_p: Optional[int] = None,
+        top_p: Optional[float] = None,
         top_k: Optional[int] = None,
         eos_early_stop: bool = False,
         minimal_topk: int = 1,