Update models/interactive_model.py

models/interactive_model.py

@@ -73,7 +73,7 @@ class BezierColorBranch(nn.Module):
 
     def create_control_points(self, x):
         x = torch.cumsum(torch.cat([torch.zeros_like(x[..., :1]), x], dim=-1), dim=-1)
-        x = torch.stack([x, torch.linspace(0, 1, steps=self.num_control_points+1).unsqueeze(0).repeat(x.shape[0], x.shape[1], 1).cuda()], dim=-1)
+        x = torch.stack([x, torch.linspace(0, 1, steps=self.num_control_points+1).unsqueeze(0).repeat(x.shape[0], x.shape[1], 1).to(x.device())], dim=-1)
         return x
 
     def forward(self, x):