Matt c7eb95581a Don't accidentally mutate the base_model_tp_plan (#36677) ...

* Don't accidentally mutate the base_model_tp_plan

* Co-authored by: Joao Gante <joaofranciscocardosogante@gmail.com>

* Trigger tests

* Marking grad accum test as slow

* Add a flaky decorator

* Add a flaky decorator

* Use cyril's codeblock

* Don't copy() when it's None

* Use cyril's new codeblock

* make fixup

2025-03-12 18:59:13 +00:00

..

__init__.py

[Test refactor 1/5] Per-folder tests reorganization (#15725)

2022-02-23 15:46:28 -05:00

test_beam_constraints.py

Generate: move generation_*.py src files into generation/*.py (#20096)

2022-11-09 15:34:08 +00:00

test_beam_search.py

Time to Say Goodbye, torch 1.7 and 1.8 (#22291)

2023-03-21 19:22:01 +01:00

test_candidate_generator.py

Universal Speculative Decoding CandidateGenerator (#35029)

2025-02-26 16:14:02 +00:00

test_configuration_utils.py

[generate] can instantiate GenerationConfig(cache_implementation="static") (#35679)

2025-01-16 17:04:54 +00:00

test_fsdp.py

Default synced_gpus to True when using FullyShardedDataParallel (#33483)

2024-10-10 14:09:04 -04:00

test_logits_process.py

use torch.testing.assertclose instead to get more details about error in cis (#35659)

2025-01-24 16:55:28 +01:00

test_stopping_criteria.py

Fix StopStringCriteria to handle tokens above len(tokenizer) (#35797)

2025-02-06 16:53:28 +00:00

test_streamers.py

Enable pytest live log and show warning logs on GitHub Actions CI runs (#35912)

2025-02-10 13:36:20 +01:00

test_utils.py

Don't accidentally mutate the base_model_tp_plan (#36677)

2025-03-12 18:59:13 +00:00