Update handler.py
Browse files- handler.py +3 -2
handler.py
CHANGED
|
@@ -24,8 +24,9 @@ class EndpointHandler:
|
|
| 24 |
"""
|
| 25 |
from transformers import AutoVideoProcessor, AutoModel
|
| 26 |
|
| 27 |
-
#
|
| 28 |
-
|
|
|
|
| 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")
|