llama : enable building vision with cmake

This commit addresses a few compilation and linking errors that were occurring when building the vision module with cmake.
danbev · Dec 16, 2024 · bb0ec7e · bb0ec7e
1 parent 78021d7
commit bb0ec7e
Show file tree

Hide file tree

Showing 3 changed files with 6 additions and 0 deletions.
diff --git a/common/CMakeLists.txt b/common/CMakeLists.txt
@@ -70,6 +70,9 @@ add_library(${TARGET} STATIC
     sampling.h
     speculative.cpp
     speculative.h
+    vision.h
+    vision.cpp
+    stb_image.h
     )
 
 if (BUILD_SHARED_LIBS)

diff --git a/examples/llava/clip.cpp b/examples/llava/clip.cpp
@@ -32,6 +32,7 @@
 #include "stb_image.h"
 
 #include <cassert>
+#include <cstdarg>
 #include <cmath>
 #include <cstdlib>
 #include <cstring>
@@ -43,6 +44,7 @@
 #include <sstream>
 #include <cinttypes>
 #include <limits>
+#include <climits>
 
 #if defined(LLAVA_LOG_OFF)
 #   define LOG_INF(...)

diff --git a/src/llama-vision.h b/src/llama-vision.h
@@ -4,6 +4,7 @@
 
 #include <vector>
 #include <array>
+#include <string>
 
 enum vision_arch {
     VISION_ARCH_UNKNOWN,