fix

app.py

@@ -285,11 +285,20 @@ def _install_aoti():
         return video, audio
     velocity._process_transformer_blocks = _types.MethodType(_proc, velocity)
     print(f"[AOTI] loaded {AOTI_REPO} + patched block loop", flush=True)
-print(f"[AOTI] base torch={torch.__version__} cuda={torch.version.cuda}", flush=True)
-try:
-    _install_aoti(); print("[AOTI] OK", flush=True)
-except Exception as _e:
-    import traceback; traceback.print_exc(); print(f"[AOTI] FAILED ({_e!r}) -> EAGER", flush=True)
+# For IC-LoRA task adapters, keep AOTI OFF by default. AOTI can replace the
+# transformer block execution with a precompiled graph that may not include the
+# currently fused IC-LoRA weights. Symptom: the output follows the reference
+# weakly / changes motion but stays grayscale, i.e. the Colorization LoRA is
+# effectively bypassed. Set USE_AOTI=1 only after verifying the LoRA effect.
+USE_AOTI = os.environ.get("USE_AOTI", "0") == "1"
+print(f"[AOTI] base torch={torch.__version__} cuda={torch.version.cuda} enabled={USE_AOTI}", flush=True)
+if USE_AOTI:
+    try:
+        _install_aoti(); print("[AOTI] OK", flush=True)
+    except Exception as _e:
+        import traceback; traceback.print_exc(); print(f"[AOTI] FAILED ({_e!r}) -> EAGER", flush=True)
+else:
+    print("[AOTI] disabled for Colorization IC-LoRA; using eager transformer so fused LoRA weights are active", flush=True)
 # ==============================================================================================
 
 
@@ -317,8 +326,11 @@ def colorize_video(video, scene_description, color_description, preset, num_fram
     # This is also the identity-safe colorization recipe recommended for this LoRA.
     gen_w, gen_h = (width * 2, height * 2) if SKIP_STAGE_2 else (width, height)
 
+    final_prompt = build_prompt(scene_description, color_description, restoration_mode)
+    print("[PROMPT]", final_prompt, flush=True)
+
     video_out, audio_out = pipeline(
-        prompt=build_prompt(scene_description, color_description, restoration_mode),
+        prompt=final_prompt,
         seed=seed, height=gen_h, width=gen_w,
         num_frames=num_frames, frame_rate=FPS,
         images=[], video_conditioning=[(ref_path, 1.0)],
@@ -338,7 +350,7 @@ with gr.Blocks(title=TITLE) as demo:
         "with the [Colorization IC-LoRA](https://huggingface.co/Lightricks/LTX-2.3-22b-IC-LoRA-Colorization)."
     )
     gr.Markdown(
-        "⚡ **Accelerated with [AOTI](https://huggingface.co/linoyts/LTX-2.3-Native-Transformer-GroupA-sm120-cu130-r20)**.  \n"
+        "⚙️ **AOTI is disabled by default for this Colorization version** so the fused IC-LoRA weights are not bypassed. Set `USE_AOTI=1` only after testing.  \n"
         "Note: the Colorization LoRA is not a true deblur/denoise/decompression/upscaling model. The old-film and high-detail options below are prompt hints only; "
         "for heavy restoration, use a dedicated deblur/decompression pass before or after colorization."
     )