Added device shift for image

hf_model.py

@@ -194,6 +194,8 @@ class Triad(nn.Module):
                               std=[0.229, 0.224, 0.225])
             ])
             image = transform(image)
+            device = next(self.parameters()).device
+            image = image.to(device) 
         embeddings = {}
         if image is not None:
             embeddings['visual_feats'] = self.visual_embedder(image)