- Using more specific LoadWeightsFailedException when a llava model…

… fails to load (#697) - Passing model path, instead of a message, to `LoadWeightsFailedException` constructor
SciSharp · Apr 26, 2024 · ecb359c · ecb359c
1 parent ccc49eb
commit ecb359c
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/LLama/Native/SafeLlamaModelHandle.cs b/LLama/Native/SafeLlamaModelHandle.cs
@@ -121,7 +121,7 @@ public static SafeLlamaModelHandle LoadFromFile(string modelPath, LLamaModelPara
                     throw new InvalidOperationException($"Model file '{modelPath}' is not readable");
 
             return llama_load_model_from_file(modelPath, lparams)
-                ?? throw new LoadWeightsFailedException($"Failed to load model {modelPath}.");
+                ?? throw new LoadWeightsFailedException(modelPath);
         }
 
         #region native API

diff --git a/LLama/Native/SafeLlavaModelHandle.cs b/LLama/Native/SafeLlavaModelHandle.cs
@@ -53,7 +53,7 @@ public static SafeLlavaModelHandle LoadFromFile(string modelPath, int verbosity
                     throw new InvalidOperationException($"Llava MMP Model file '{modelPath}' is not readable");
 
             return clip_model_load(modelPath, verbosity)
-                ?? throw new RuntimeError($"Failed to load LLaVa model {modelPath}.");          
+                ?? throw new LoadWeightsFailedException(modelPath);
         }
 
         /// <summary>