change link to download weights

segment-anything-models-java · Oct 24, 2024 · 5001acd · 5001acd
1 parent 404bec3
commit 5001acd
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java b/src/main/java/ai/nets/samj/install/EfficientViTSamEnvManager.java
@@ -125,7 +125,7 @@ else if (!PlatformDetection.getArch().equals(PlatformDetection.ARCH_ARM64) && !P
 	 * URL to download the EfficientViTSAM model. It needs to be used with String.format(EVITSAM_URL, "l0"), whre l0 could be any of 
 	 * the existing EfficientVitSAM model 
 	 */
-	final static private String EVITSAM_URL = "https://huggingface.co/han-cai/efficientvit-sam/resolve/main/%s.pt?download=true";
+	final static private String EVITSAM_URL = "https://huggingface.co/mit-han-lab/efficientvit-sam/resolve/main/efficientvit_sam_%s.pt";
 
 	private EfficientViTSamEnvManager(String modelType) {
 		List<String> modelTypes = EFFICIENTVITSAM_BYTE_SIZES_MAP.keySet().stream().collect(Collectors.toList());