From d8f62f9edbbbaf8ab6a09ca1d67c7e1e74a3bd47 Mon Sep 17 00:00:00 2001
From: JingyaHuang <huang_jingya@outlook.com>
Date: Wed, 13 Aug 2025 16:42:21 +0000
Subject: [PATCH 1/4] fix: dummy cache context

---
 optimum/neuron/modeling_diffusion.py | 6 ++++++
 1 file changed, 6 insertions(+)

diff --git a/optimum/neuron/modeling_diffusion.py b/optimum/neuron/modeling_diffusion.py
index 76c07ed02..d6d0ede61 100644
--- a/optimum/neuron/modeling_diffusion.py
+++ b/optimum/neuron/modeling_diffusion.py
@@ -21,6 +21,7 @@
 import shutil
 from abc import abstractmethod
 from collections import OrderedDict
+from contextlib import contextmanager
 from dataclasses import asdict
 from pathlib import Path
 from tempfile import TemporaryDirectory
@@ -1321,6 +1322,11 @@ def forward(
             outputs = (outputs,)
         return outputs
 
+    # Dummy context manager
+    @contextmanager
+    def cache_context(self, name: str):
+        yield
+
 
 class NeuronModelVaeEncoder(_NeuronDiffusionModelPart):
     def __init__(

From 61802163e83872e88a42cf38544d761681a907ea Mon Sep 17 00:00:00 2001
From: JingyaHuang <huang_jingya@outlook.com>
Date: Wed, 20 Aug 2025 07:50:15 +0000
Subject: [PATCH 2/4] bump diffusers

---
 pyproject.toml | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 0624bfb38..4244f6e32 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -61,7 +61,7 @@ tests = [
     "sentencepiece",
     "datasets",
     "sacremoses",
-    "diffusers>=0.31.0, <=0.34.0",
+    "diffusers==0.35.*",
     "safetensors",
     "sentence-transformers >= 2.2.0",
     "peft==0.16.0",
@@ -106,7 +106,7 @@ neuronx = [
     "libneuronxla==2.2.4410.0",
 ]
 diffusers = [
-    "diffusers>=0.31.0, <=0.34.0",
+    "diffusers==0.35.*",
     "peft==0.16.0",
 ]
 sentence-transformers = [

From 256247dfa083f70c61276d4e1d23f7fe7ff8a86f Mon Sep 17 00:00:00 2001
From: JingyaHuang <huang_jingya@outlook.com>
Date: Wed, 20 Aug 2025 07:59:00 +0000
Subject: [PATCH 3/4] bump peft, peft>=0.17.0 is required

---
 pyproject.toml | 6 +++---
 1 file changed, 3 insertions(+), 3 deletions(-)

diff --git a/pyproject.toml b/pyproject.toml
index 4244f6e32..6154db31a 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -64,7 +64,7 @@ tests = [
     "diffusers==0.35.*",
     "safetensors",
     "sentence-transformers >= 2.2.0",
-    "peft==0.16.0",
+    "peft==0.17.0",
     "trl==0.11.4",
     "compel",
     "rjieba",
@@ -83,7 +83,7 @@ quality = [
 ]
 training = [
     "trl == 0.11.4",
-    "peft == 0.16.0",
+    "peft == 0.17.0",
     "evaluate == 0.4.3",
     "neuronx_distributed==0.13.14393",
 ]
@@ -107,7 +107,7 @@ neuronx = [
 ]
 diffusers = [
     "diffusers==0.35.*",
-    "peft==0.16.0",
+    "peft==0.17.0",
 ]
 sentence-transformers = [
     "sentence-transformers >= 2.2.0",

From 4947fabddd498fa6fa39f23d6c9f4d748f853159 Mon Sep 17 00:00:00 2001
From: JingyaHuang <huang_jingya@outlook.com>
Date: Wed, 20 Aug 2025 10:17:18 +0000
Subject: [PATCH 4/4] fix: pin torchcodec version

---
 pyproject.toml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/pyproject.toml b/pyproject.toml
index 6154db31a..dbb926091 100644
--- a/pyproject.toml
+++ b/pyproject.toml
@@ -75,7 +75,7 @@ tests = [
     "mediapipe",
     "timm >= 1.0.0",
     "hf_transfer",
-    "torchcodec",
+    "torchcodec < 0.6.0",
 ]
 quality = [
     "ruff",