From 7ba43f9f451bdc96067535c960161d82877b12e9 Mon Sep 17 00:00:00 2001
From: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
Date: Sun, 21 Sep 2025 12:40:46 +0100
Subject: [PATCH 1/3] Removing instantiation in forward

Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
---
 monai/networks/nets/diffusion_model_unet.py | 3 ---
 1 file changed, 3 deletions(-)

diff --git a/monai/networks/nets/diffusion_model_unet.py b/monai/networks/nets/diffusion_model_unet.py
index 3e496979bf..eb12777fd6 100644
--- a/monai/networks/nets/diffusion_model_unet.py
+++ b/monai/networks/nets/diffusion_model_unet.py
@@ -2063,9 +2063,6 @@ def forward(
         h = h.reshape(h.shape[0], -1)
 
         # 5. out
-        self.out = nn.Sequential(
-            nn.Linear(h.shape[1], 512), nn.ReLU(), nn.Dropout(0.1), nn.Linear(512, self.out_channels)
-        )
         output: torch.Tensor = self.out(h)
 
         return output

From 66f1b3a4866c16e0c502f702279992ef8c2dd2af Mon Sep 17 00:00:00 2001
From: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
Date: Sun, 21 Sep 2025 13:09:17 +0100
Subject: [PATCH 2/3] Type fix

Signed-off-by: Eric Kerfoot <17726042+ericspod@users.noreply.github.com>
---
 monai/networks/nets/diffusion_model_unet.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/monai/networks/nets/diffusion_model_unet.py b/monai/networks/nets/diffusion_model_unet.py
index eb12777fd6..41cb5b2b9d 100644
--- a/monai/networks/nets/diffusion_model_unet.py
+++ b/monai/networks/nets/diffusion_model_unet.py
@@ -2016,7 +2016,7 @@ def __init__(
 
         last_dim_flattened = int(reduce(lambda x, y: x * y, input_shape) * channels[-1])
 
-        self.out: Optional[nn.Module] = nn.Sequential(
+        self.out: nn.Module = nn.Sequential(
             nn.Linear(last_dim_flattened, 512), nn.ReLU(), nn.Dropout(0.1), nn.Linear(512, self.out_channels)
         )
 

From 99c130c394023f3e2bafbb04edb5e34c0aa126cb Mon Sep 17 00:00:00 2001
From: "pre-commit-ci[bot]"
 <66853113+pre-commit-ci[bot]@users.noreply.github.com>
Date: Sun, 21 Sep 2025 12:09:51 +0000
Subject: [PATCH 3/3] [pre-commit.ci] auto fixes from pre-commit.com hooks

for more information, see https://pre-commit.ci
---
 monai/networks/nets/diffusion_model_unet.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/monai/networks/nets/diffusion_model_unet.py b/monai/networks/nets/diffusion_model_unet.py
index 41cb5b2b9d..cb0c69d033 100644
--- a/monai/networks/nets/diffusion_model_unet.py
+++ b/monai/networks/nets/diffusion_model_unet.py
@@ -34,7 +34,6 @@
 import math
 from collections.abc import Sequence
 from functools import reduce
-from typing import Optional
 
 import numpy as np
 import torch