MoCha: experimental context windows support

2025-10-21 18:32:40 +03:00
parent 56120d633e
commit 4e36aee658
2 changed files with 19 additions and 5 deletions
@@ -167,7 +167,7 @@ class MochaEmbeds:
            "mocha_embeds": mocha_embeds,
            "num_frames": F,
            "target_shape": target_shape,
-            "num_refs": num_refs,
+            "mocha_num_refs": num_refs,
        }
        
        return (image_embeds,)