Failing to import ONNX with error: "No variable with name".

[ArtificialCactus]

Describe the bug
I was trying to import this model (I upgraded its ONNX version to 16 using the script provided in the guide) and after writing mpk file burn-import panics with this error:

...
DEBUG burn_import::burn::graph: Registering node => 'gemm'
 INFO burn_core::record::file: File exists, replacing
DEBUG burn_import::burn::graph: Building the scope nodes len => '257'
ERROR burn_import::logger: PANIC => panicked at C:/<...>/cargo\registry\src\index.crates.io-1949cf8c6b5b557f\burn-import-0.18.0\src\burn\scope.rs:43:13:
No variable with name PartitionedCall_PartitionedCall_transpose_243_0

  --- stderr

thread 'main' panicked at C:/<...>/cargo\registry\src\index.crates.io-1949cf8c6b5b557f\burn-import-0.18.0\src\burn\scope.rs:43:13:
No variable with name PartitionedCall_PartitionedCall_transpose_243_0

To Reproduce
Follow the official guide from the Burn Book. My build.rs file:

use burn_import::onnx::ModelGen;

fn main() {
    ModelGen::new()
        .input("models/discogs_track_embeddings-effnet-bs64-1-v16.onnx")
        .out_dir("models/out")
        .run_from_script();
}

Additional context
I checked the model using onnx.checker.check_model() in Python and it didn't return any errors.

[antimora]

The bug seems to be fixed recently on main. Please try again. I can successfully convert.