From 5083501d1f98d6a2e9009316804f8d00235ad3cf Mon Sep 17 00:00:00 2001
From: Shivam Tripathi <119807222+blackytyphoon@users.noreply.github.com>
Date: Mon, 30 Sep 2024 12:17:28 +0530
Subject: [PATCH] Update diffusion.py

I replaced self.first_stage_model.decoder with self.first_stage_model.decode in the instance check, as per your request.
---
 sgm/models/diffusion.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/sgm/models/diffusion.py b/sgm/models/diffusion.py
index 2f3efd3c..8d86850a 100644
--- a/sgm/models/diffusion.py
+++ b/sgm/models/diffusion.py
@@ -123,7 +123,7 @@ def decode_first_stage(self, z):
         all_out = []
         with torch.autocast("cuda", enabled=not self.disable_first_stage_autocast):
             for n in range(n_rounds):
-                if isinstance(self.first_stage_model.decoder, VideoDecoder):
+                if isinstance(self.first_stage_model.decode, VideoDecoder):
                     kwargs = {"timesteps": len(z[n * n_samples : (n + 1) * n_samples])}
                 else:
                     kwargs = {}