Spaces:

Jiaqi-hkust
/

hawk

Runtime error

Jiaqi-hkust commited on Feb 25

Commit

7bae158

verified ·

1 Parent(s): 78e208d

Update hawk/conversation/conversation_video.py

Files changed (1) hide show

hawk/conversation/conversation_video.py CHANGED Viewed

@@ -172,7 +172,7 @@ conv_llava_llama_2 = Conversation(
 class Chat:
     def __init__(self, model, vis_processor, device='cuda:0'):
-        self.device = "cpu"
         self.model = model
         self.vis_processor = vis_processor
         self.image_vis_processor = Blip2ImageEvalProcessor()

 class Chat:
     def __init__(self, model, vis_processor, device='cuda:0'):
+        self.device = "cuda"
         self.model = model
         self.vis_processor = vis_processor
         self.image_vis_processor = Blip2ImageEvalProcessor()