From 128dc20f8839d68e5eceee9a7d33ba5488effe1f Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 31 May 2024 11:44:05 +0300
Subject: [PATCH] Update nodes.py

---
 nodes.py | 7 +++++--
 1 file changed, 5 insertions(+), 2 deletions(-)

diff --git a/nodes.py b/nodes.py
index a69e4b7..6ca8986 100644
--- a/nodes.py
+++ b/nodes.py
@@ -507,8 +507,11 @@ class ToonCrafterI2V:
 
             ## reconstruct from latent to pixel space
             self.model.first_stage_model.to(device)
-            additional_decode_kwargs = {'ref_context': hs}
-            decoded_images = self.model.decode_first_stage(samples) #b c t h w
+            if mm.XFORMERS_IS_AVAILABLE:
+                additional_decode_kwargs = {'ref_context': hs}
+                decoded_images = self.model.decode_first_stage(samples, **additional_decode_kwargs) #b c t h w
+            else:
+                decoded_images = self.model.decode_first_stage(samples) #b c t h w
             self.model.first_stage_model.to('cpu')
         
             video = decoded_images.detach().cpu()