diff --git a/mlx/io/load.h b/mlx/io/load.h index e75057f68..3a58aff59 100644 --- a/mlx/io/load.h +++ b/mlx/io/load.h @@ -2,12 +2,17 @@ #pragma once -#include -#include -#include #include #include +#include +#ifdef _MSC_VER +#include +#else +#include +#include +#endif + #include "mlx/io/threadpool.h" namespace mlx::core {