diff --git a/demo.ipynb b/demo.ipynb index 1d2f469..2067ebc 100644 --- a/demo.ipynb +++ b/demo.ipynb @@ -613,6 +613,9 @@ " ).strip()\n", "\n", " # === Part B: Video2LoRA Inference (zero visual tokens) ===\n", + " # Re-patch the base model layers with the LoRA forward function before binding weights\n", + " model.patch_lora_forward()\n", + "\n", " apply_lora_to_layers(\n", " model.base_model,\n", " model.hypernet.layer_indices,\n",