Merge pull request #34 from dhruvarayasam/file-not-found-fix

File not found error handling in loadModel method in GGUF.java

Author: mikepapadim

external/tornadovm

@@ -1,18 +1,21 @@
 package com.example.core.model;
 
 import com.example.auxiliary.Timer;
+import com.example.core.model.GGUF.GGUFTensorInfo;
 import com.example.core.model.tensor.FloatTensor;
 import com.example.core.model.tensor.GGMLTensorEntry;
 import com.example.core.types.MetadataValueType;
 import com.example.core.types.Pair;
 
+import java.io.FileNotFoundException;
 import java.io.IOException;
 import java.lang.foreign.Arena;
 import java.lang.foreign.MemorySegment;
 import java.nio.ByteBuffer;
 import java.nio.ByteOrder;
 import java.nio.channels.FileChannel;
 import java.nio.charset.StandardCharsets;
+import java.nio.file.Files;
 import java.nio.file.Path;
 import java.util.HashMap;
 import java.util.List;
@@ -36,10 +39,19 @@ public final class GGUF {
     private long tensorDataOffset;
 
     public static GGUF loadModel(Path modelPath) throws IOException {
+
+        // file existence check
+        if (!Files.exists(modelPath)) {
+            throw new FileNotFoundException("Model file not found: " + modelPath);
+        }
+
+        // second check to make sure that nothing goes wrong during model loading
         try (FileChannel fileChannel = FileChannel.open(modelPath); var ignored = Timer.log("Parse " + modelPath)) {
             GGUF gguf = new GGUF();
             gguf.loadModelImpl(fileChannel);
             return gguf;
+        } catch (Exception e) {
+            throw new RuntimeException("Unexpected error while loading GGUF model from " + modelPath, e);
         }
     }
 

src/main/java/com/example/core/model/GGUF.java

@@ -67,7 +67,7 @@ private static ModelType detectModelType(Map<String, Object> metadata) {
 
     public static Model loadModel(Path ggufPath, int contextLength, boolean loadWeights) throws IOException {
         // initial load of metadata from gguf file
-        GGUF gguf = GGUF.loadModel(ggufPath);
+        GGUF gguf = GGUF.loadModel(ggufPath); 
         FileChannel fileChannel = FileChannel.open(ggufPath, StandardOpenOption.READ);
         // detect model type
         ModelType modelType = detectModelType(gguf.getMetadata());