monkeypatch.trainer_accelerator_args
monkeypatch.trainer_accelerator_args
allow adding additional kwargs to Accelerator init
Functions
| Name | Description |
|---|---|
| patch_create_accelerate_code_for_fp8 | Monkeypatch create_accelerator_and_postprocess so it checks for additional kwargs. |
patch_create_accelerate_code_for_fp8
monkeypatch.trainer_accelerator_args.patch_create_accelerate_code_for_fp8(
enable_fsdp_float8_all_gather,
)Monkeypatch create_accelerator_and_postprocess so it checks for additional kwargs.