rf inversion fixes

2024-12-13 01:04:33 +02:00
parent 3a00471a85
commit 1307c1fe31
2 changed files with 1 additions and 3 deletions
@@ -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]

@@ -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()