From abebd113befc427dae39856c108176fa851bef33 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 31 Dec 2024 12:31:30 -0800 Subject: [PATCH] Update __init__.py --- unsloth/__init__.py | 4 +--- 1 file changed, 1 insertion(+), 3 deletions(-) diff --git a/unsloth/__init__.py b/unsloth/__init__.py index 25d4e2b0a5..0b46794e9d 100644 --- a/unsloth/__init__.py +++ b/unsloth/__init__.py @@ -58,9 +58,7 @@ pass # And optimize pinning of memory os.environ["PYTORCH_CUDA_ALLOC_CONF"] = \ "expandable_segments:True,"\ - "roundup_power2_divisions:[64:128,256:64,>:32],"\ - "pinned_use_cuda_host_register:True,"\ - "pinned_num_register_threads:8" + "roundup_power2_divisions:[64:128,256:64,>:32]" # Hugging Face Hub faster downloads if "HF_HUB_ENABLE_HF_TRANSFER" not in os.environ: