Skip to content

Commit

Permalink
Update tests/test_tuners_utils.py
Browse files Browse the repository at this point in the history
  • Loading branch information
ltoniazzi authored Aug 25, 2024
1 parent 8087f48 commit a2f7354
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/test_tuners_utils.py
Original file line number Diff line number Diff line change
Expand Up @@ -1163,7 +1163,7 @@ def assert_warning_triggered(records, warn_start, warn_end, triggered=True):
triggered=False,
)

# No warning when loading model with tie_word_embeddings although but not relevant target module
# No warning when loading model with tie_word_embeddings but not relevant target module
with warnings.catch_warnings(record=True) as records_inject_tied_no_target:
model_not_tied = get_peft_model(
AutoModelForCausalLM.from_pretrained(model_id, tie_word_embeddings=False),
Expand Down

0 comments on commit a2f7354

Please sign in to comment.