diff --git a/stable-diffusion.cpp b/stable-diffusion.cpp
index fb26aab..c0f9db7 100644
--- a/stable-diffusion.cpp
+++ b/stable-diffusion.cpp
@@ -10,6 +10,8 @@
 #include <string>
 #include <unordered_map>
 #include <vector>
+#include <cstring>
+#include <iterator>
 
 #include "ggml/ggml.h"
 #include "stable-diffusion.h"