update app.py

open_flamingo/open_flamingo/src/flamingo_lm.py

@@ -92,7 +92,9 @@ class FlamingoLayer(nn.Module):
             elif not self.training:
                 if self.add_visual_token:
                     if self.input_ids is None:
+                        print(decoder_layer_kwargs)
                         self.input_ids = decoder_layer_kwargs["input_ids"]
+
                     else:
                         self.input_ids = torch.cat([self.input_ids, decoder_layer_kwargs["input_ids"]], dim=-1)
                     visual_token_position = (self.input_ids[..., -1] == self.visual_token_id).nonzero().reshape(-1)