diff --git a/pyproject.toml b/pyproject.toml index eb61d50d88..ed89ce5337 100644 --- a/pyproject.toml +++ b/pyproject.toml @@ -55,7 +55,7 @@ huggingfacenotorch = [ "huggingface_hub>=0.34.0", "hf_transfer", "diffusers", - "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,<=4.57.2", + "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,<=4.57.3", "trl>=0.18.2,!=0.19.0,<=0.24.0", ] huggingface = [ @@ -526,7 +526,7 @@ colab-new = [ "unsloth_zoo>=2025.11.6", "packaging", "tyro", - "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,<=4.57.2", + "transformers>=4.51.3,!=4.52.0,!=4.52.1,!=4.52.2,!=4.52.3,!=4.53.0,!=4.54.0,!=4.55.0,!=4.55.1,!=4.57.0,<=4.57.3", "datasets>=3.4.1,!=4.0.*,!=4.1.0,<4.4.0", "sentencepiece>=0.2.0", "tqdm", diff --git a/unsloth/models/vision.py b/unsloth/models/vision.py index 3b02850115..ed19f587cf 100644 --- a/unsloth/models/vision.py +++ b/unsloth/models/vision.py @@ -672,10 +672,12 @@ class FastBaseModel: model.fast_generate = model.generate model.fast_generate_batches = error_out_no_vllm if offload_embedding: - if bool(os.environ.get("WSL_DISTRO_NAME") or os.environ.get("WSL_INTEROP")): + if bool( + os.environ.get("WSL_DISTRO_NAME") or os.environ.get("WSL_INTEROP") + ): # WSL doesn't work with offloaded embeddings pass - elif os.name == 'nt': + elif os.name == "nt": # Windows doesn't work with offloaded embeddings pass else: