20fb766ad2
T5-Gemma outputs BFloat16 which numpy does not support. Cast all feature tensors with .float() before .numpy(). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>