jiang commited on
Commit
08384c6
·
1 Parent(s): 36c6351
Files changed (1) hide show
  1. demo.py +1 -0
demo.py CHANGED
@@ -230,6 +230,7 @@ def visual_grounding(image, text):
230
  with torch.autocast("cpu"), torch.no_grad():
231
  if isinstance(models, list):
232
  model = models[0]
 
233
  min_len = 6
234
  max_len = 210
235
  model.eval()
 
230
  with torch.autocast("cpu"), torch.no_grad():
231
  if isinstance(models, list):
232
  model = models[0]
233
+ model = model.float()
234
  min_len = 6
235
  max_len = 210
236
  model.eval()