Update hawk/models/video_llama.py

hawk/models/video_llama.py

@@ -403,7 +403,7 @@ class VideoLLAMA(Blip2Base):
 
     def encode_videoQformer_visual(self, image, motion=False):
         if motion is False:
-            device = "cuda:0"
+            device = "cuda"
             
             # input shape b,c,t,h,w
             batch_size,_,time_length,_,_ = image.size()
@@ -454,7 +454,7 @@ class VideoLLAMA(Blip2Base):
         
         else:
             # Motion Encoder
-            device = "cuda:0"
+            device = "cuda"
             
             # input shape b,c,t,h,w
             batch_size,_,time_length,_,_ = image.size()