24.4 MB
puar-playground's picture
auto device = CUDA -> CPU (MPS opt-in); document MPS is slower than CPU for this tiny model
d436f2f verified