Nguyen Thai Thao Uyen commited on
Commit
81a19fa
·
1 Parent(s): 2485872

run.py Fix cache permission issue

Browse files
Files changed (1) hide show
  1. run.py +2 -2
run.py CHANGED
@@ -13,8 +13,8 @@ def pred(src):
13
 
14
  # -- load model configuration
15
  MODEL_FILE = "sam_model.pth"
16
- model_config = SamConfig.from_pretrained("facebook/sam-vit-base")
17
- processor = SamProcessor.from_pretrained("facebook/sam-vit-base")
18
 
19
  model = SamModel(config=model_config)
20
  model.load_state_dict(torch.load(MODEL_FILE))
 
13
 
14
  # -- load model configuration
15
  MODEL_FILE = "sam_model.pth"
16
+ model_config = SamConfig.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
17
+ processor = SamProcessor.from_pretrained("facebook/sam-vit-base", cache_dir=cache_dir)
18
 
19
  model = SamModel(config=model_config)
20
  model.load_state_dict(torch.load(MODEL_FILE))