Update app.py
Browse files
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
|
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 |
|