How to use google/videoprism-large-f8r288 with VideoPrism:
# Install from https://github.com/google-deepmind/videoprism import jax from videoprism import models as vp flax_model = vp.get_model("google/videoprism-large-f8r288") loaded_state = vp.load_pretrained_weights("google/videoprism-large-f8r288") @jax.jit def forward_fn(inputs, train=False): return flax_model.apply(loaded_state, inputs, train=train)
MHRDYN7