jhj0517 commited on
Commit
5c1c241
·
2 Parent(s): 511b7dd 968457a

Merge branch 'master' into huggingface

Browse files
Files changed (1) hide show
  1. modules/sam_inference.py +1 -0
modules/sam_inference.py CHANGED
@@ -90,6 +90,7 @@ class SamInference:
90
  logger.exception("Error while loading SAM2 model for video predictor")
91
 
92
  try:
 
93
  self.model = build_sam2(
94
  config_file=config_name,
95
  ckpt_path=model_path,
 
90
  logger.exception("Error while loading SAM2 model for video predictor")
91
 
92
  try:
93
+ self.video_predictor = None
94
  self.model = build_sam2(
95
  config_file=config_name,
96
  ckpt_path=model_path,