[Fix] Fix lint error

Author: HAOCHENYE

xtuner/v1/module/rms_norm/rms_norm.py

@@ -28,7 +28,7 @@ def forward(self, hidden_states: torch.Tensor) -> torch.Tensor:
         # hidden_states = hidden_states * torch.rsqrt(variance + self.variance_epsilon)
         # return (weight * hidden_states).to(input_dtype)  # gpt_oss
         # return weight * hidden_states.to(input_dtype)  # Llama
-        return rms_norm(hidden_states, weight, epsilon=self.variance_epsilon)  # xtuner
+        return rms_norm(hidden_states, weight, epsilon=self.variance_epsilon)  # type: ignore[operator]
 
     def init_weights(self):
         self.weight.data.fill_(1.0)

xtuner/v1/train/trainer.py

@@ -54,6 +54,7 @@
 
 from .toy_tokenizer import UTF8ByteTokenizer
 
+
 # TODO: Move DEVICE to `xtuner.utils.device`
 DEVICE = get_device()
 DEVICE_MODULE = get_torch_device_module()