jhj0517 commited on
Commit
968457a
1 Parent(s): 7dc0c85

offload video predictor

Browse files
Files changed (1) hide show
  1. 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,