Update models/unet.py

models/unet.py

@@ -842,6 +842,8 @@ class MotionCLR(nn.Module):
         print(device)
         print(next(self.clip_model.parameters()).device)
         print("00000000")
+        self.clip_model = self.clip_model.to(device)
+        print("00000000")
         with torch.no_grad():
             texts = clip.tokenize(raw_text, truncate=True).to(
                 device