correct new name of file

segment-anything-models-java · Oct 24, 2024 · 363fe2d · 363fe2d
1 parent 5001acd
commit 363fe2d
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
@@ -235,7 +235,7 @@ public boolean checkModelWeightsInstalled() {
 		if (!EfficientViTSamJ.getListOfSupportedEfficientViTSAM().contains(modelType))
 			throw new IllegalArgumentException("The provided model is not one of the supported EfficientViT models: " 
 												+ EfficientViTSamJ.getListOfSupportedEfficientViTSAM());
-		File weightsFile = Paths.get(this.path, "envs", EVITSAM_ENV_NAME, EVITSAM_NAME, "weights", modelType + ".pt").toFile();
+		File weightsFile = Paths.get(this.path, "envs", EVITSAM_ENV_NAME, EVITSAM_NAME, "weights", "efficientvit_sam_" + modelType + ".pt").toFile();
 		if (!weightsFile.isFile()) return false;
 		if (weightsFile.length() != EFFICIENTVITSAM_BYTE_SIZES_MAP.get(modelType)) return false;
 		return true;