[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
038ea7907f
commit
3f5815cc02
1 changed files with 1 additions and 6 deletions
|
|
@ -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"
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue