robot0820 commited on
Commit
375f86a
Β·
verified Β·
1 Parent(s): e9fe7b0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -41,7 +41,7 @@ def generate_answer(image, text):
41
  force_batchify=True
42
  ).to(vl_gpt.device)
43
 
44
- prepare_inputs = prepare_inputs.to_dict()
45
  prepare_inputs = {k: (v.to(torch.float16) if torch.is_tensor(v) else v)
46
  for k, v in prepare_inputs.items()}
47
 
 
41
  force_batchify=True
42
  ).to(vl_gpt.device)
43
 
44
+ prepare_inputs = dict(prepare_inputs)
45
  prepare_inputs = {k: (v.to(torch.float16) if torch.is_tensor(v) else v)
46
  for k, v in prepare_inputs.items()}
47