This commit is contained in:
Dan Saunders
2025-03-14 01:58:07 +00:00
parent a6ef6c7764
commit e1a02a32b5

View File

@@ -125,8 +125,8 @@ def normalize_config(cfg):
with open(ds_config_path, encoding="utf-8") as f:
cfg.deepspeed = json.load(f)
if cfg.sequence_parallel_size is None:
cfg.sequence_parallel_size = 1
if cfg.sequence_parallel_degree is None:
cfg.sequence_parallel_degree = 1
if cfg.saves_per_epoch:
save_steps = 1.0 / (cfg.saves_per_epoch * cfg.num_epochs)