Spaces:
Runtime error
Runtime error
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
|