From 3a00471a850faf0641b4e89c55b993f23361863b Mon Sep 17 00:00:00 2001
From: kijai <40791699+kijai@users.noreply.github.com>
Date: Fri, 13 Dec 2024 01:01:46 +0200
Subject: [PATCH] Update nodes_rf_inversion.py

---
 nodes_rf_inversion.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/nodes_rf_inversion.py b/nodes_rf_inversion.py
index 343a5b7..1dc20a4 100644
--- a/nodes_rf_inversion.py
+++ b/nodes_rf_inversion.py
@@ -85,6 +85,7 @@ class HyVideoInverseSampler:
     CATEGORY = "HunyuanVideoWrapper"
 
     def process(self, model, hyvid_embeds, flow_shift, steps, embedded_guidance_scale, seed, samples, gamma, start_step, end_step, gamma_trend, force_offload):
+        model = model.model
         device = mm.get_torch_device()
         offload_device = mm.unet_offload_device()
         dtype = model["dtype"]
@@ -287,6 +288,7 @@ class HyVideoReSampler:
 
     def process(self, model, hyvid_embeds, flow_shift, steps, embedded_guidance_scale, 
                 samples, inversed_latents, force_offload, start_step, end_step, eta_base, eta_trend):
+        model = model.model
         device = mm.get_torch_device()
         offload_device = mm.unet_offload_device()
         dtype = model["dtype"]