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,
|