diff --git a/models/wan/wan_handler.py b/models/wan/wan_handler.py
index 9aecfd1..10b1902 100644
--- a/models/wan/wan_handler.py
+++ b/models/wan/wan_handler.py
@@ -55,7 +55,7 @@ class family_handler():
         "cfg_zero" : True,
         "cfg_star" : True,
         "adaptive_projected_guidance" : True,  
-        "skip_steps_cache" : True,
+        "skip_steps_cache" : not (base_model_type in ["i2v_2_2", "ti2v_2_2" ] or "URLs2" in model_def),
         })
 
         return extra_model_def
diff --git a/wgp.py b/wgp.py
index 614819e..a5f77ce 100644
--- a/wgp.py
+++ b/wgp.py
@@ -4270,7 +4270,7 @@ def generate_video(
         if prompt_enhancer_image_caption_model != None and prompt_enhancer !=None and len(prompt_enhancer)>0:
             text_encoder_max_tokens = 256
             send_cmd("progress", [0, get_latest_status(state, "Enhancing Prompt")])
-            from ltx_video.utils.prompt_enhance_utils import generate_cinematic_prompt
+            from models.ltx_video.utils.prompt_enhance_utils import generate_cinematic_prompt
             prompt_images = []
             if "I" in prompt_enhancer:
                 if image_start != None: