peterproofpath commited on
Commit
c8b5767
·
verified ·
1 Parent(s): 0f76ce8

Update handler.py

Browse files
Files changed (1) hide show
  1. handler.py +3 -2
handler.py CHANGED
@@ -24,8 +24,9 @@ class EndpointHandler:
24
  """
25
  from transformers import AutoVideoProcessor, AutoModel
26
 
27
- # Use the model path provided by the endpoint, or default to HF hub
28
- model_id = path if path else "facebook/vjepa2-vitl-fpc64-256"
 
29
 
30
  # Determine device
31
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 
24
  """
25
  from transformers import AutoVideoProcessor, AutoModel
26
 
27
+ # Always load from the official Facebook model on HuggingFace Hub
28
+ # (path points to /repository which is our custom handler, not the model weights)
29
+ model_id = "facebook/vjepa2-vitl-fpc64-256"
30
 
31
  # Determine device
32
  self.device = torch.device("cuda" if torch.cuda.is_available() else "cpu")