From 93f38c4f8535c834ed13a89d5af41a8ffc006d75 Mon Sep 17 00:00:00 2001
From: zjt <937178071@qq.com>
Date: Sat, 21 Oct 2023 22:07:46 +0800
Subject: [PATCH] change large model dir

---
 backend.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/backend.py b/backend.py
index b571520..59449bc 100644
--- a/backend.py
+++ b/backend.py
@@ -12,7 +12,7 @@ print(torch.cuda.is_available())
 # checkpoint = "sam_vit_b_01ec64.pth"
 print(time.time())
 # set large model related configs
-checkpoint = "notebooks/sam_vit_h_4b8939.pth"
+checkpoint = "sam_vit_h_4b8939.pth"
 model_type = "vit_h"
 sam = sam_model_registry[model_type](checkpoint=checkpoint)
 sam.to(device='cuda')