419 kB
ViuAI's picture
Fix CUDA OOM on T4: cast model to amp_dtype directly, batch_size=2, safe RMSNorm
9d0f9ca verified