diff --git a/backends/arm/tosa/backend.py b/backends/arm/tosa/backend.py
index 42af22ff09f..1631526e360 100644
--- a/backends/arm/tosa/backend.py
+++ b/backends/arm/tosa/backend.py
@@ -133,7 +133,14 @@ def _preprocess(  # noqa: C901
         if not artifact_path:
             artifact_path = ""
 
-        tosa_graph = ts.TosaSerializer(artifact_path)
+        version = tosa_spec.version
+        tosa_graph = ts.TosaSerializer(
+            artifact_path,
+            targetMajor=version.major,
+            targetMinor=version.minor,
+            targetPatch=version.micro,
+            targetDraft=False,
+        )
 
         if not (
             tosa_spec.version.major == ts.TOSA_VERSION_MAJOR
diff --git a/examples/arm/setup.sh b/examples/arm/setup.sh
index db32c0c416f..a11b4a2eebd 100755
--- a/examples/arm/setup.sh
+++ b/examples/arm/setup.sh
@@ -290,7 +290,9 @@ if [[ $is_script_sourced -eq 0 ]]; then
     fi
 
     pushd tosa-tools
+    git fetch origin main
     git checkout 8468d041c50c6d806f3c1c18c66d7ef641e46580 # serialization lib pybindings
+    git cherry-pick 368f0cd745b2a1569bf36f077daeba95775de192 # perf fix for >2gb models
     if [[ ! -d "reference_model" ]]; then
         log_step "main" "[error] Missing reference_model directory in tosa-tools repo."
         exit 1