[pre-commit.ci] auto fixes from pre-commit.com hooks
for more information, see https://pre-commit.ci
This commit is contained in:
parent
f33762fab7
commit
c4dc0e94b6
2 changed files with 10 additions and 2 deletions
|
|
@ -3073,8 +3073,12 @@ class FastLlamaModel:
|
|||
clean_gpu_cache()
|
||||
|
||||
import warnings as _w
|
||||
|
||||
with _w.catch_warnings():
|
||||
_w.filterwarnings("ignore", message=".*target_parameters.*were set but no parameter was matched.*")
|
||||
_w.filterwarnings(
|
||||
"ignore",
|
||||
message = ".*target_parameters.*were set but no parameter was matched.*",
|
||||
)
|
||||
model = _get_peft_model(model, lora_config)
|
||||
# Fix LoraConfig.auto_mapping is None
|
||||
fix_lora_auto_mapping(model)
|
||||
|
|
|
|||
|
|
@ -1210,8 +1210,12 @@ class FastBaseModel:
|
|||
use_gradient_checkpointing = use_gradient_checkpointing,
|
||||
)
|
||||
import warnings as _w
|
||||
|
||||
with _w.catch_warnings():
|
||||
_w.filterwarnings("ignore", message=".*target_parameters.*were set but no parameter was matched.*")
|
||||
_w.filterwarnings(
|
||||
"ignore",
|
||||
message = ".*target_parameters.*were set but no parameter was matched.*",
|
||||
)
|
||||
model = _get_peft_model(model, lora_config)
|
||||
# Apply QAT + LoRA if specified
|
||||
if qat_scheme is not None:
|
||||
|
|
|
|||
Loading…
Add table
Add a link
Reference in a new issue