Fix the signature

torch-ext/torch_binding.cpp

@@ -5,7 +5,7 @@
 
 
 TORCH_LIBRARY_EXPAND(TORCH_EXTENSION_NAME, ops) {
-    ops.def("residual_rms(Tensor input, Tensor residual, Tensor weight, Tensor scale_tensor, double epsilon, Tensor! output, Tensor next_buffer, int64_t num_threads, bool force_scalar) -> ()");
+    ops.def("residual_rms(Tensor input, Tensor residual, Tensor weight, Tensor scale_tensor, float epsilon, Tensor! output, Tensor next_buffer, int num_threads, bool force_scalar) -> ()");
     ops.impl("residual_rms", torch::kCUDA, &residual_rms);
 }