update demo

sp.py

@@ -14,10 +14,10 @@ class SimPool(nn.Module):
         self.wk = nn.Linear(dim, dim, bias=qkv_bias)
         
         if gamma is not None:
-            self.gamma = torch.tensor([gamma], device='cuda')
+            self.gamma = torch.tensor([gamma])
             if use_beta:
-                self.beta = nn.Parameter(torch.tensor([0.0], device='cuda'))
-        self.eps = torch.tensor([1e-6], device='cuda')
+                self.beta = nn.Parameter(torch.tensor([0.0]))
+        self.eps = torch.tensor([1e-6])
 
         self.gamma = gamma
         self.use_beta = use_beta