From 1b2cb292fb64fe9fc024981483d9a8464d36ba64 Mon Sep 17 00:00:00 2001
From: "Jennifer J. Sun" <40725770+jenjsun@users.noreply.github.com>
Date: Tue, 15 Jul 2025 18:17:47 -0700
Subject: [PATCH 1/2] Update model-libraries-snippets.ts

---
 packages/tasks/src/model-libraries-snippets.ts | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
index 9782c3bfea..a069ff5566 100644
--- a/packages/tasks/src/model-libraries-snippets.ts
+++ b/packages/tasks/src/model-libraries-snippets.ts
@@ -1559,7 +1559,7 @@ import jax
 import jax.numpy as jnp
 from videoprism import models as vp
 
-flax_model = vp.MODELS["${model.id}"]()
+flax_model = vp.get_model("${model.id}")
 loaded_state = vp.load_pretrained_weights("${model.id}")
 
 @jax.jit

From 14413bbd2e400f7902f6fe1ad326948de8342b01 Mon Sep 17 00:00:00 2001
From: "Jennifer J. Sun" <40725770+jenjsun@users.noreply.github.com>
Date: Fri, 18 Jul 2025 07:51:13 -0700
Subject: [PATCH 2/2] Rename videoprism function call in
 model-libraries-snippets.ts

---
 packages/tasks/src/model-libraries-snippets.ts | 1 -
 1 file changed, 1 deletion(-)

diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts
index 2c3e535eea..f023ce1cef 100644
--- a/packages/tasks/src/model-libraries-snippets.ts
+++ b/packages/tasks/src/model-libraries-snippets.ts
@@ -1587,7 +1587,6 @@ image = sana(
 export const videoprism = (model: ModelData): string[] => [
 	`# Install from https://github.com/google-deepmind/videoprism
 import jax
-import jax.numpy as jnp
 from videoprism import models as vp
 
 flax_model = vp.get_model("${model.id}")