Spaces:
Running
Running
jiang
commited on
Commit
·
08384c6
1
Parent(s):
36c6351
update
Browse files
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()
|