From e43babe76fce73e3d0d3d2040f80cef911d6c6cb Mon Sep 17 00:00:00 2001
From: Daniel Han <danielhanchen@gmail.com>
Date: Fri, 20 Jun 2025 06:13:45 -0700
Subject: [PATCH] Update __init__.py

---
 unsloth/__init__.py | 6 +++++-
 1 file changed, 5 insertions(+), 1 deletion(-)

diff --git a/unsloth/__init__.py b/unsloth/__init__.py
index 58c2a19814..fdaf7b483a 100644
--- a/unsloth/__init__.py
+++ b/unsloth/__init__.py
@@ -60,9 +60,13 @@ keynames = "\n" + "\n".join(os.environ.keys())
 if "HF_XET_HIGH_PERFORMANCE" not in os.environ:
     os.environ["HF_XET_HIGH_PERFORMANCE"] = "1"
 pass
-if "\nCOLAB_" in keynames:
+# Disable XET cache sine it eats too much space
+if "HF_XET_CHUNK_CACHE_SIZE_BYTES" not in os.environ:
     os.environ["HF_XET_CHUNK_CACHE_SIZE_BYTES"] = "0"
 pass
+if "\nCOLAB_" in keynames:
+    os.environ["HF_XET_RECONSTRUCT_WRITE_SEQUENTIALLY"] = "0"
+pass
 
 # Log Unsloth is being used
 os.environ["UNSLOTH_IS_PRESENT"] = "1"