Return image hidden states (#33426)

Commit

1 year ago

Return image hidden states (#33426) * fix * return image hidden states * fix copies * fix test