Jiaqi-hkust commited on
Commit
b2ed81d
·
verified ·
1 Parent(s): a44c0f5

Update hawk/conversation/conversation_video.py

Browse files
hawk/conversation/conversation_video.py CHANGED
@@ -307,8 +307,8 @@ class Chat:
307
  video = self.vis_processor.transform(video)
308
  video_motion = self.vis_processor.transform(video_motion)
309
 
310
- video = video.unsqueeze(0).to(self.device).clone().detach()
311
- video_motion = video_motion.unsqueeze(0).to(self.device).clone().detach()
312
  # print(image)
313
  else:
314
  raise NotImplementedError
 
307
  video = self.vis_processor.transform(video)
308
  video_motion = self.vis_processor.transform(video_motion)
309
 
310
+ video = video.unsqueeze(0).to(self.device).clone().detach().half()
311
+ video_motion = video_motion.unsqueeze(0).to(self.device).clone().detach().half()
312
  # print(image)
313
  else:
314
  raise NotImplementedError