[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
e73632f841
commit
4e8ce86041
1 changed files with 1 additions and 0 deletions
|
|
@ -1403,6 +1403,7 @@ class FastSentenceTransformer(FastModel):
|
|||
qat_scheme = kwargs.get("qat_scheme", None)
|
||||
if qat_scheme is not None:
|
||||
from ._utils import _prepare_model_for_qat
|
||||
|
||||
peft_model = _prepare_model_for_qat(peft_model, qat_scheme)
|
||||
|
||||
# Determine compile mode (only if not using gradient checkpointing)
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue