Fail if candle cannot load prostt5

steineggerlab · Jun 13, 2024 · 488888f · 488888f
1 parent 740b0f1
commit 488888f
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/strucclustutils/structcreatedb.cpp b/src/strucclustutils/structcreatedb.cpp
@@ -301,6 +301,10 @@ int structcreatedb(int argc, const char **argv, const Command& command) {
             return EXIT_FAILURE;
         }
         ProstT5 *model = prostt5_load(modelWeights.c_str(), false, par.gpu == 0, false, quantized);
+        if (model == NULL) {
+            // error message is already printed by prostt5_load
+            return EXIT_FAILURE;
+        }
 #ifdef OPENMP
         size_t localThreads = par.gpu != 0 ? 1 : par.threads;
 #endif