Update model.py

model.py

@@ -381,6 +381,7 @@ class StreamMultiDiffusion(nn.Module):
         question = 'Question: What are in the image? Answer:'
         inputs = self.i2t_processor(image, question, return_tensors='pt')
         out = self.i2t_model.generate(**{k: v.to(self.i2t_model.device) for k, v in inputs.items()}, max_new_tokens=77)
+        print(out)
         prompt = self.i2t_processor.decode(out[0], skip_special_tokens=True).strip()
         return prompt