From 0f5b470c04438e8c0ffdf11abeada952a6fa29b6 Mon Sep 17 00:00:00 2001
From: Concedo <39025047+LostRuins@users.noreply.github.com>
Date: Thu, 30 Mar 2023 09:28:04 +0800
Subject: [PATCH] more library checks

---
 llamacpp_for_kobold.py | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/llamacpp_for_kobold.py b/llamacpp_for_kobold.py
index 94ab32376..987c662a3 100644
--- a/llamacpp_for_kobold.py
+++ b/llamacpp_for_kobold.py
@@ -286,8 +286,8 @@ def RunServerMultiThreaded(addr, port, embedded_kailite = None):
 
 def main(args): 
     global use_blas
-    if not os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "libopenblas.dll")):
-        print("Warning: libopenblas.dll not found. OpenBLAS will be disabled.")
+    if not os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "libopenblas.dll")) or not os.path.exists(os.path.join(os.path.dirname(os.path.realpath(__file__)), "llamacpp_blas.dll")):
+        print("Warning: libopenblas.dll or llamacpp_blas.dll not found. OpenBLAS will be disabled.")
         use_blas = False
     elif not args.noblas:
         print("Attempting to use OpenBLAS library for faster prompt ingestion. A compatible libopenblas.dll will be required.")