Skip to content

Commit 6e3ba3a

Browse files
committed
added intern model test
Signed-off-by: Onkar Chougule <[email protected]>
1 parent eb7db29 commit 6e3ba3a

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

tests/transformers/models/test_image_text_to_text_intern.py

Lines changed: 3 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -203,7 +203,9 @@ def __call__(self, pixel_values,
203203

204204

205205
import ipdb; ipdb.set_trace()
206-
generated_ids_aic = model.generate(inputs=orig_inputs, generation_len=128)
206+
exec_info = model.generate(inputs=inputs, generation_len=128)
207+
print(exec_info)
208+
generated_ids_aic = exec_info.generated_ids
207209
print(generated_ids_aic)
208210
generated_texts = tokenizer.batch_decode(generated_ids_aic, skip_special_tokens=True)
209211
print(generated_texts)

0 commit comments

Comments
 (0)