diff --git a/model.h b/model.h
index 865c9b0..3b51222 100644
--- a/model.h
+++ b/model.h
@@ -6,6 +6,7 @@
 #include <memory>
 #include <string>
 #include <vector>
+#include <set>
 
 #include "ggml/ggml-backend.h"
 #include "ggml/ggml.h"