From de35c441fcc6301a43d1572aceb7b4225ea45e0b Mon Sep 17 00:00:00 2001
From: KiritoDv <kiritodev01@gmail.com>
Date: Wed, 31 Jan 2024 12:45:31 -0600
Subject: [PATCH] Fixed linux issues

---
 src/utils/Decompressor.h | 1 +
 src/utils/TorchUtils.h   | 1 +
 2 files changed, 2 insertions(+)

diff --git a/src/utils/Decompressor.h b/src/utils/Decompressor.h
index 4e3c298..5551b57 100644
--- a/src/utils/Decompressor.h
+++ b/src/utils/Decompressor.h
@@ -4,6 +4,7 @@
 #include <cstdint>
 #include <unordered_map>
 #include <yaml-cpp/yaml.h>
+#include <optional>
 
 enum class CompressionType {
     MIO0,
diff --git a/src/utils/TorchUtils.h b/src/utils/TorchUtils.h
index 330c7fa..e3d89e6 100644
--- a/src/utils/TorchUtils.h
+++ b/src/utils/TorchUtils.h
@@ -3,6 +3,7 @@
 #include <string>
 #include <fstream>
 #include <sstream>
+#include <iomanip>
 
 namespace Torch {
 template< typename T >