Update Serialization.cpp

Used to resolve Windows 10 crash issues.

Update Serialization.cpp
Used to resolve Windows 10 crash issues.
e9618ab9 · PengXin · GitHub · 8e3bab93 · e9618ab9
Unverified Commit e9618ab9 authored Mar 06, 2025 by PengXin Committed by GitHub Mar 06, 2025
Hide whitespace changes
Inline Side-by-side

Showing with 2 additions and 0 deletions

src/Serialization.cpp src/Serialization.cpp +2 -0

No files found.
--- a/src/Serialization.cpp
+++ b/src/Serialization.cpp
@@ -106,7 +106,9 @@ SafeTensors::SafeTensors(const std::string &filename) {
 }

 SafeTensors::~SafeTensors() {
+#ifndef _WIN32  
    checkCUDA(cudaHostUnregister(const_cast<char *>(this->mapped->data())));
+#endif
 }

 void SafeTensors::parseHeader() {