Add default value to "checkpoint_folder" in "load_state_dict" of bf16_optimizer (#6446)

Add default value `checkpoint_folder=None` for compatibility.

Co-authored-by: Olatunji Ruwase <olruwase@microsoft.com>
This commit is contained in:
Jiancheng Liu 2024-09-04 16:32:02 -04:00 коммит произвёл GitHub
Родитель 5d1a30c033
Коммит ddd3571823
Не найден ключ, соответствующий данной подписи
Идентификатор ключа GPG: B5690EEEBB952194
1 изменённых файлов: 1 добавлений и 1 удалений

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

@ -482,7 +482,7 @@ class BF16_Optimizer(ZeROOptimizer):
def load_state_dict(self,
state_dict_list,
checkpoint_folder,
checkpoint_folder=None,
load_optimizer_states=True,
load_from_fp32_weights=False,
load_serial=None,