Use `json_schema_extra` instead of extra keyword in `Field` (#6764)

> Using extra keyword arguments on `Field` is deprecated and will be
removed. Use `json_schema_extra` instead. (Extra keys: 'new_param').
Deprecated in Pydantic V2.0 to be removed in V3.0. See Pydantic V2
Migration Guide at https://errors.pydantic.dev/2.9/migration/

Co-authored-by: Logan Adams <114770087+loadams@users.noreply.github.com>
This commit is contained in:
Quentin Gallouédec 2024-11-20 02:04:47 +01:00 коммит произвёл GitHub
Родитель 065398d5de
Коммит 83e4364fbd
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 1 добавлений и 1 удалений

Просмотреть файл

@ -311,7 +311,7 @@ class DeepSpeedZeroConfig(DeepSpeedConfigModel):
for efficient all_2_all_reduce comm
"""
mics_shard_size: int = Field(-1, new_param="mics_shard_size")
mics_shard_size: int = Field(-1, json_schema_extra={"new_param": "mics_shard_size"})
mics_hierarchical_params_gather: bool = False