Spaces:
Runtime error
Runtime error
jhj0517
commited on
Commit
·
968457a
1
Parent(s):
7dc0c85
offload video predictor
Browse files- modules/sam_inference.py +1 -0
modules/sam_inference.py
CHANGED
|
@@ -86,6 +86,7 @@ class SamInference:
|
|
| 86 |
logger.exception("Error while loading SAM2 model for video predictor")
|
| 87 |
|
| 88 |
try:
|
|
|
|
| 89 |
self.model = build_sam2(
|
| 90 |
config_file=config_name,
|
| 91 |
ckpt_path=model_path,
|
|
|
|
| 86 |
logger.exception("Error while loading SAM2 model for video predictor")
|
| 87 |
|
| 88 |
try:
|
| 89 |
+
self.video_predictor = None
|
| 90 |
self.model = build_sam2(
|
| 91 |
config_file=config_name,
|
| 92 |
ckpt_path=model_path,
|