From 4ec1b216e01bdaa01a470b094bd36589f5bcb583 Mon Sep 17 00:00:00 2001
From: Daniel Han-Chen <danielhanchen@gmail.com>
Date: Mon, 26 Feb 2024 14:15:37 +1100
Subject: [PATCH] Update save.py

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

diff --git a/unsloth/save.py b/unsloth/save.py
index 83e13bd51c..51ddeb30e7 100644
--- a/unsloth/save.py
+++ b/unsloth/save.py
@@ -369,6 +369,7 @@ def unsloth_save_model(
 
     # Switch to our fast saving modules if it's a slow PC!
     n_cpus = psutil.cpu_count(logical = False)
+    if n_cpus is None: n_cpus = 1
 
     if safe_serialization is None:
         safe_serialization = True
@@ -669,7 +670,9 @@ def save_to_gguf(
         pass
     pass
 
-    n_cpus = psutil.cpu_count()*2
+    n_cpus = psutil.cpu_count()
+    if n_cpus is None: n_cpus = 1
+    n_cpus *= 2
     # Concurrency from https://rentry.org/llama-cpp-conversions#merging-loras-into-a-model
     
     final_location = f"./{model_directory}-unsloth.{first_conversion.upper()}.gguf"