fix: correct VideoPrism import (videoprism.models, not videoprism); add flax dep
videoprism/__init__.py is empty — API lives in videoprism.models. Fix: from videoprism import models as vp (not import videoprism as vp). Also add flax to managed venv packages (required by videoprism Flax model). Co-Authored-By: Claude Sonnet 4.6 <noreply@anthropic.com>
This commit is contained in:
@@ -68,12 +68,11 @@ class FeaturesUtils:
|
||||
def _ensure_videoprism(self):
|
||||
if self._vp_model is not None:
|
||||
return
|
||||
import videoprism as vp
|
||||
from videoprism import models as vp
|
||||
import jax
|
||||
print("[FeaturesUtils] Loading VideoPrism...")
|
||||
self._vp_model = vp.get_model("videoprism_public_v1_base")
|
||||
self._vp_state = vp.load_pretrained_weights("videoprism_public_v1_base")
|
||||
import jax
|
||||
self._jax_forward = jax.jit(
|
||||
lambda x: self._vp_model.apply(self._vp_state, x, train=False)
|
||||
)
|
||||
|
||||
Reference in New Issue
Block a user