From 3a28446a54b4a967b1452c94f32b3e96991bb3df Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 17 Mar 2026 21:32:38 -0700 Subject: [PATCH 1/6] Trim ~255 MB of unused packages from Studio setup (#4395) * Comment out large unused packages from Studio setup requirements Audited all packages installed by `unsloth studio setup` against actual imports in unsloth, unsloth_zoo, and studio/backend. The following have zero imports anywhere and are the largest offenders by disk size: - gradio (148 MB) in studio.txt -- Studio uses React + FastAPI, not Gradio - executorch (41.5 MB) in extras-no-deps.txt -- no imports found - scikit-learn (31.8 MB) in extras.txt -- no imports found - MeCab (19.9 MB) in extras.txt -- Japanese tokenizer, no imports found - coremltools (10.2 MB) in extras.txt -- Apple CoreML, no imports found - uroman (4.0 MB) in extras.txt -- romanization tool, no imports found Total savings: ~255 MB (~32% of the 805 MB installed by setup). Each line is commented out with the package size annotated so they can be re-enabled easily if needed in the future. * Restore scikit-learn -- needed by sentence_transformers sentence_transformers is installed with --no-deps in extras-no-deps.txt, so its sklearn dependency is not auto-resolved. Multiple modules in sentence_transformers import sklearn at the top level (evaluation, util/similarity), so removing scikit-learn would break embedding jobs. --- studio/backend/requirements/extras-no-deps.txt | 2 +- studio/backend/requirements/extras.txt | 8 ++++---- studio/backend/requirements/studio.txt | 2 +- 3 files changed, 6 insertions(+), 6 deletions(-) diff --git a/studio/backend/requirements/extras-no-deps.txt b/studio/backend/requirements/extras-no-deps.txt index 29bd421a0a..3bdce81cc1 100644 --- a/studio/backend/requirements/extras-no-deps.txt +++ b/studio/backend/requirements/extras-no-deps.txt @@ -8,7 +8,7 @@ snac # TRL and related packages trl==0.23.1 git+https://github.com/meta-pytorch/OpenEnv.git -executorch>=1.0.1 +# executorch>=1.0.1 # 41.5 MB - no imports in unsloth/zoo/studio torch-c-dlpack-ext sentence_transformers==5.2.0 transformers==4.57.1 diff --git a/studio/backend/requirements/extras.txt b/studio/backend/requirements/extras.txt index 51ef69cf5d..170b1c9ec8 100644 --- a/studio/backend/requirements/extras.txt +++ b/studio/backend/requirements/extras.txt @@ -4,7 +4,7 @@ tomli-w # ExecuTorch dependencies ruamel.yaml -coremltools +# coremltools # 10.2 MB - Apple CoreML, no imports in unsloth/zoo/studio expecttest flatbuffers hydra-core @@ -15,7 +15,7 @@ pytest<9.0 pytest-json-report pytest-rerunfailures==15.1 pytest-xdist -# Also needed by sentence_transformers +# Also needed by sentence_transformers (installed with --no-deps in extras-no-deps.txt) scikit-learn==1.7.1 # Additional extras @@ -26,8 +26,8 @@ omegaconf einx pyloudnorm openai-whisper -uroman -MeCab +# uroman # 4.0 MB - romanization, no imports found +# MeCab # 19.9 MB - Japanese tokenizer, no imports found loguru flatten_dict ffmpy diff --git a/studio/backend/requirements/studio.txt b/studio/backend/requirements/studio.txt index da0b4f382b..186ba82fe0 100644 --- a/studio/backend/requirements/studio.txt +++ b/studio/backend/requirements/studio.txt @@ -10,7 +10,7 @@ datasets==4.3.0 pyjwt easydict addict -gradio>=4.0.0 +# gradio>=4.0.0 # 148 MB - Studio uses React + FastAPI, not Gradio huggingface-hub==0.36.2 structlog>=24.1.0 diceware From f3f52e2d84f281d5574ddda3bd226807d7bca489 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 17 Mar 2026 22:07:21 -0700 Subject: [PATCH 2/6] Use blobless clone in README install instructions (#4403) Reduces clone size from ~50MB to ~5MB by skipping blobs that are no longer in the current tree but still in git history. --- README.md | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/README.md b/README.md index 692887088e..dae34e12c3 100644 --- a/README.md +++ b/README.md @@ -58,7 +58,7 @@ Use our [Docker image](https://hub.docker.com/r/unsloth/unsloth) ```unsloth/unsl You can also install directly from source: ``` -git clone https://github.com/unslothai/unsloth.git +git clone --filter=blob:none https://github.com/unslothai/unsloth.git cd unsloth pip install -e . unsloth studio setup From 9232126734a2a8a7088edb1f52b9c71082f38ced Mon Sep 17 00:00:00 2001 From: Lee Jackson <130007945+Imagineer99@users.noreply.github.com> Date: Wed, 18 Mar 2026 05:39:51 +0000 Subject: [PATCH 3/6] fix(studio): use explicit Cancel for model load toast (#4377) --- .../features/chat/components/model-load-status.tsx | 11 +++++------ 1 file changed, 5 insertions(+), 6 deletions(-) diff --git a/studio/frontend/src/features/chat/components/model-load-status.tsx b/studio/frontend/src/features/chat/components/model-load-status.tsx index ad9946b0fe..afec327a0f 100644 --- a/studio/frontend/src/features/chat/components/model-load-status.tsx +++ b/studio/frontend/src/features/chat/components/model-load-status.tsx @@ -4,7 +4,6 @@ import { Progress } from "@/components/ui/progress"; import { Spinner } from "@/components/ui/spinner"; import { Button } from "@/components/ui/button"; -import { XIcon } from "lucide-react"; type ModelLoadDescriptionProps = { title?: string | null; @@ -28,11 +27,11 @@ export function ModelLoadDescription({ const hasProgress = typeof progressPercent === "number"; return ( -
+
-
+
{title ?

{title}

: null} {hasProgress ? (
@@ -49,13 +48,13 @@ export function ModelLoadDescription({ {onStop ? ( ) : null}
From 24753290ba5b109e9240b92f90900756ee321337 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 17 Mar 2026 22:50:55 -0700 Subject: [PATCH 4/6] Update README.md --- README.md | 6 ++++-- 1 file changed, 4 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index dae34e12c3..2f7dd9c748 100644 --- a/README.md +++ b/README.md @@ -50,7 +50,8 @@ Unsloth Studio works on **Windows, Linux, WSL** and **macOS**. #### Windows, MacOS, Linux or WSL: ``` -pip install unsloth +pip install --upgrade pip && pip install uv +uv pip install unsloth --torch-backend=auto unsloth studio setup unsloth studio -H 0.0.0.0 -p 8888 ``` @@ -58,9 +59,10 @@ Use our [Docker image](https://hub.docker.com/r/unsloth/unsloth) ```unsloth/unsl You can also install directly from source: ``` +pip install --upgrade pip && pip install uv git clone --filter=blob:none https://github.com/unslothai/unsloth.git cd unsloth -pip install -e . +uv pip install -e . --torch-backend=auto unsloth studio setup unsloth studio -H 0.0.0.0 -p 8888 ``` From 767c31b0e084fc1188a0e3e35a09123303fc0984 Mon Sep 17 00:00:00 2001 From: Daniel Han Date: Tue, 17 Mar 2026 22:53:11 -0700 Subject: [PATCH 5/6] Update README.md --- README.md | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index 2f7dd9c748..faf2bec141 100644 --- a/README.md +++ b/README.md @@ -70,13 +70,14 @@ unsloth studio -H 0.0.0.0 -p 8888 ### Unsloth Core (code-based) #### Windows, Linux, WSL ```bash -pip install unsloth +pip install --upgrade pip && pip install uv +uv pip install unsloth --torch-backend=auto ``` For Windows, `pip install unsloth` works only if you have Pytorch installed. Read our [Windows Guide](https://unsloth.ai/docs/get-started/install/windows-installation). You can use the same Docker image as Unsloth Studio. #### AMD, Intel -For RTX 50x, B200, 6000 GPUs: `pip install unsloth`. Read our guides for: [Blackwell](https://unsloth.ai/docs/blog/fine-tuning-llms-with-blackwell-rtx-50-series-and-unsloth) and [DGX Spark](https://unsloth.ai/docs/blog/fine-tuning-llms-with-nvidia-dgx-spark-and-unsloth).
+For RTX 50x, B200, 6000 GPUs: `uv pip install unsloth --torch-backend=auto`. Read our guides for: [Blackwell](https://unsloth.ai/docs/blog/fine-tuning-llms-with-blackwell-rtx-50-series-and-unsloth) and [DGX Spark](https://unsloth.ai/docs/blog/fine-tuning-llms-with-nvidia-dgx-spark-and-unsloth).
To install Unsloth on **AMD** and **Intel** GPUs, follow our [AMD Guide](https://unsloth.ai/docs/get-started/install/amd) and [Intel Guide](https://unsloth.ai/docs/get-started/install/intel). ## ✨ Free Notebooks From 67d3519cab9fde8603d89bd6303864e6ca852cb8 Mon Sep 17 00:00:00 2001 From: Michael Han <107991372+shimmyshimmer@users.noreply.github.com> Date: Tue, 17 Mar 2026 23:04:54 -0700 Subject: [PATCH 6/6] Update README.md --- README.md | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/README.md b/README.md index faf2bec141..e60b1f98d9 100644 --- a/README.md +++ b/README.md @@ -44,8 +44,10 @@ Unsloth can be used in two ways: through **[Unsloth Studio](https://unsloth.ai/d Unsloth Studio works on **Windows, Linux, WSL** and **macOS**. * **CPU:** Supported for **chat inference only** -* **NVIDIA GPUs:** Training works on RTX 30/40/50, Blackwell, DGX Spark, DGX Station and more +* **NVIDIA:** Training works on RTX 30/40/50, Blackwell, DGX Spark, Station and more * **macOS:** Currently supports chat only; **MLX training** is coming very soon +* **AMD:** Chat works. Train with [Unsloth Core](#unsloth-core-code-based). Studio support is coming soon. +* **Coming soon:** Training support for Apple MLX, AMD, and Intel. * **Multi-GPU:** Available now, with a major upgrade on the way #### Windows, MacOS, Linux or WSL: