Update chatNT.py

chatNT.py

@@ -1188,6 +1188,11 @@ class RotaryEmbeddingBis(torch.nn.Module):
             heads[..., heads.shape[-1] // 2 :],
         )
 
+        print("x_first device : ", x_first.device)
+        print("cos device : ", cos.device)
+        print("x_second device : ", x_second.device)
+        print("sin device : ", sin.device)
+
         first_part = x_first * cos - x_second * sin
         second_part = x_second * cos + x_first * sin