From ca940896ae1316e777b07dea7f49248a1d91ba27 Mon Sep 17 00:00:00 2001
From: Mr-Neutr0n <64578610+Mr-Neutr0n@users.noreply.github.com>
Date: Wed, 11 Feb 2026 19:47:35 +0530
Subject: [PATCH] fix: add missing n_frames attribute in StandardDiffusionLoss

---
 sgm/modules/diffusionmodules/loss.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/sgm/modules/diffusionmodules/loss.py b/sgm/modules/diffusionmodules/loss.py
index c397c6585..0e000a5e2 100644
--- a/sgm/modules/diffusionmodules/loss.py
+++ b/sgm/modules/diffusionmodules/loss.py
@@ -17,6 +17,7 @@ def __init__(
         loss_type: str = "l2",
         offset_noise_level: float = 0.0,
         batch2model_keys: Optional[Union[str, List[str]]] = None,
+        n_frames: Optional[int] = None,
     ):
         super().__init__()
 
@@ -27,6 +28,7 @@ def __init__(
 
         self.loss_type = loss_type
         self.offset_noise_level = offset_noise_level
+        self.n_frames = n_frames
 
         if loss_type == "lpips":
             self.lpips = LPIPS().eval()