Viu-1.5B-MoE / data /scripts
319 kB
ViuAI's picture
Fix CUDA OOM on T4: cast model to amp_dtype directly, batch_size=2, safe RMSNorm
41a0763 verified