nekoshadow commited on
Commit
fe0626e
1 Parent(s): 67cb1f8

Fix import issue

Browse files
Files changed (1) hide show
  1. sam_utils.py +1 -1
sam_utils.py CHANGED
@@ -7,7 +7,7 @@ import time
7
  from segment_anything import sam_model_registry, SamPredictor
8
 
9
  def sam_init(device_id=0):
10
- sam_checkpoint = os.path.join(os.path.dirname(__file__), "ckpt/sam_vit_h_4b8939.pth")
11
  model_type = "vit_h"
12
 
13
  device = "cuda:{}".format(device_id) if torch.cuda.is_available() else "cpu"
 
7
  from segment_anything import sam_model_registry, SamPredictor
8
 
9
  def sam_init(device_id=0):
10
+ sam_checkpoint = os.path.join(os.path.dirname(__file__), "SyncDreamer/ckpt/sam_vit_h_4b8939.pth")
11
  model_type = "vit_h"
12
 
13
  device = "cuda:{}".format(device_id) if torch.cuda.is_available() else "cpu"