Update lora_w2w.py

lora_w2w.py

@@ -95,10 +95,10 @@ class LoRAModule(nn.Module):
         del self.org_module
 
     def forward(self, x):
-        print(self.proj.device)
-        print(self.v1.device)
-        print(self.mean1.device)
-        print(self.std1.device)
+        print(x.dtype)
+        print(self.proj.dtype)
+        print(self.std1.dtype)
+        print(self.mean1.dtype)
         return self.org_forward(x) +\
             (x@(([email protected])*self.std1+self.mean1).T)@((([email protected])*self.std2+self.mean2))*self.multiplier*self.scale