From 1709beba5b2f9ff53dcde07743321d7a90eba82a Mon Sep 17 00:00:00 2001
From: Zhao Changmin <changmin.zhao@intel.com>
Date: Tue, 10 Oct 2023 14:57:23 +0800
Subject: [PATCH] LLM: Explicitly close pickle file pointer before removing
 temporary directory (#9120)

* fp close
---
 python/llm/src/bigdl/llm/utils/lazy_load_torch.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/python/llm/src/bigdl/llm/utils/lazy_load_torch.py b/python/llm/src/bigdl/llm/utils/lazy_load_torch.py
index eda2d6de..5de30b94 100644
--- a/python/llm/src/bigdl/llm/utils/lazy_load_torch.py
+++ b/python/llm/src/bigdl/llm/utils/lazy_load_torch.py
@@ -179,6 +179,7 @@ def lazyload(
                       f"but get {pickle_paths}. ")
     pickle_fp = zf.open(pickle_paths[0], 'r')
     state_dict = _load(pickle_fp, None, pickle, pickle_file=pickle_paths[0][:-4], zip_file=zf)
+    fp.close()  # Otherwise on windows this may be marked as reading
     return state_dict