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()
|