Update models/unet.py

models/unet.py

@@ -856,7 +856,8 @@ class MotionCLR(nn.Module):
             log_attn=self.vis_attn,
             edit_config=edit_config,
         )
-        
+
+        self.clip_model = self.clip_model.cuda()
         self.embed_text = self.embed_text.cuda()
         self.textTransEncoder = self.textTransEncoder.cuda()
         self.text_ln = self.text_ln.cuda()