From 1307c1fe3116ec02aea8bb54bc9a20ff812ab620 Mon Sep 17 00:00:00 2001 From: kijai <40791699+kijai@users.noreply.github.com> Date: Fri, 13 Dec 2024 01:04:33 +0200 Subject: [PATCH] rf inversion fixes --- nodes.py | 2 +- nodes_rf_inversion.py | 2 -- 2 files changed, 1 insertion(+), 3 deletions(-) diff --git a/nodes.py b/nodes.py index 443aa5a..0d35f67 100644 --- a/nodes.py +++ b/nodes.py @@ -923,7 +923,7 @@ class HyVideoSampler: target_width = align_to(width, 16) freqs_cos, freqs_sin = get_rotary_pos_embed( - model["pipe"].transformer, num_frames, target_height, target_width + transformer, num_frames, target_height, target_width ) n_tokens = freqs_cos.shape[0] diff --git a/nodes_rf_inversion.py b/nodes_rf_inversion.py index 1dc20a4..3f07e9f 100644 --- a/nodes_rf_inversion.py +++ b/nodes_rf_inversion.py @@ -133,7 +133,6 @@ class HyVideoInverseSampler: elif model["manual_offloading"]: transformer.to(device) - mm.unload_all_models() mm.soft_empty_cache() gc.collect() @@ -334,7 +333,6 @@ class HyVideoReSampler: elif model["manual_offloading"]: transformer.to(device) - mm.unload_all_models() mm.soft_empty_cache() gc.collect()