Skip to content

Commit 5e4b47c

Browse files
authored
Update VideoPrism function call name in model-libraries-snippets.ts (#1643)
We updated the some functions in VideoPrism to make the model loading work with model.id from the huggingface repo; this is a small update to videoprism inside model-libraries-snippets.ts so the code still runs
1 parent 04a3d77 commit 5e4b47c

File tree

1 file changed

+1
-2
lines changed

1 file changed

+1
-2
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 1 addition & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -1644,10 +1644,9 @@ image = sana(
16441644
export const videoprism = (model: ModelData): string[] => [
16451645
`# Install from https://github.com/google-deepmind/videoprism
16461646
import jax
1647-
import jax.numpy as jnp
16481647
from videoprism import models as vp
16491648
1650-
flax_model = vp.MODELS["${model.id}"]()
1649+
flax_model = vp.get_model("${model.id}")
16511650
loaded_state = vp.load_pretrained_weights("${model.id}")
16521651
16531652
@jax.jit

0 commit comments

Comments
 (0)