Can't load example MNIST model

[castrong]

Hi all - I'm trying to use ONNX for the first time and I'm getting an error when I try to load the example MNIST model (see screenshot below).

I'd appreciate any advice people have on resolving this! Thank you!

[ayush-1506]

Looks like you're on an older version of Flux. Could you try using the latest one?