finhdev commited on
Commit
e53312a
·
verified ·
1 Parent(s): 01f2a47

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +1 -1
handler.py CHANGED
@@ -48,7 +48,7 @@ class EndpointHandler:
48
  img_feat = self.model.get_image_features(**img_in)
49
 
50
  img_feat = img_feat / img_feat.norm(dim=-1, keepdim=True)
51
- txt_feat = txt_feat / txt_feat.norm(dim=-1, keepdim=True)
52
 
53
  img_feat = img_feat.float() # ← add these two lines
54
  txt_feat = txt_feat.float() # ←
 
48
  img_feat = self.model.get_image_features(**img_in)
49
 
50
  img_feat = img_feat / img_feat.norm(dim=-1, keepdim=True)
51
+ # txt_feat = txt_feat / txt_feat.norm(dim=-1, keepdim=True)
52
 
53
  img_feat = img_feat.float() # ← add these two lines
54
  txt_feat = txt_feat.float() # ←