diff --git a/unsloth/_gpu_init.py b/unsloth/_gpu_init.py index d7a7b2ca6e..b35e18cf75 100644 --- a/unsloth/_gpu_init.py +++ b/unsloth/_gpu_init.py @@ -241,6 +241,7 @@ if "is_bf16_supported" not in globals(): def is_bf16_supported(including_emulation = False): return SUPPORTS_BFLOAT16 + # For Gradio HF Spaces? # if "SPACE_AUTHOR_NAME" not in os.environ and "SPACE_REPO_NAME" not in os.environ: import triton