From d57b2742ab7bebd922fa81b8855ee7365834c02e Mon Sep 17 00:00:00 2001 From: Roland Tannous Date: Wed, 18 Feb 2026 09:21:53 +0000 Subject: [PATCH] renamed UNSLOTH_FLEX_ATTENTION to UNSLOTH_ENABLE_FLEX_ATTENTION --- studio/backend/main.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/studio/backend/main.py b/studio/backend/main.py index e027998dcf..e7a33750ff 100644 --- a/studio/backend/main.py +++ b/studio/backend/main.py @@ -34,7 +34,7 @@ async def lifespan(app: FastAPI): props = torch.cuda.get_device_properties(0) sm_version = props.major * 10 + props.minor if sm_version >= 120: - os.environ["UNSLOTH_FLEX_ATTENTION"] = "0" + os.environ["UNSLOTH_ENABLE_FLEX_ATTENTION"] = "0" import logging logging.getLogger(__name__).info( f"GPU sm_{sm_version} detected — setting UNSLOTH_FLEX_ATTENTION=0"