diff --git a/unsloth/models/sentence_transformer.py b/unsloth/models/sentence_transformer.py index f6657023d2..bd5add6729 100644 --- a/unsloth/models/sentence_transformer.py +++ b/unsloth/models/sentence_transformer.py @@ -932,12 +932,7 @@ class FastSentenceTransformer(FastModel): load_in_fp8 = False load_in_16bit = False - if ( - int(load_in_4bit) - + int(load_in_8bit) - + int(load_in_16bit) - >= 2 - ): + if int(load_in_4bit) + int(load_in_8bit) + int(load_in_16bit) >= 2: raise RuntimeError( "Unsloth: Can only load in 4bit or 8bit or 16bit, not a combination!\n" "Also, we by default set `load_in_16bit = True`.\n"