fix dp arg
This commit is contained in:
@@ -840,7 +840,14 @@ class EngineArgs:
|
|||||||
# Get the list of attributes of this dataclass.
|
# Get the list of attributes of this dataclass.
|
||||||
attrs = [attr.name for attr in dataclasses.fields(cls)]
|
attrs = [attr.name for attr in dataclasses.fields(cls)]
|
||||||
# Set the attributes from the parsed arguments.
|
# Set the attributes from the parsed arguments.
|
||||||
engine_args = cls(**{attr: getattr(args, attr) for attr in attrs})
|
# Fields that exist in the dataclass but have no corresponding CLI
|
||||||
|
# argument (e.g. data_parallel_size, enable_expert_parallel) will
|
||||||
|
# fall back to their dataclass defaults (typically read from env vars).
|
||||||
|
engine_args = cls(**{
|
||||||
|
attr: getattr(args, attr)
|
||||||
|
for attr in attrs
|
||||||
|
if hasattr(args, attr)
|
||||||
|
})
|
||||||
return engine_args
|
return engine_args
|
||||||
|
|
||||||
def create_model_config(self) -> ModelConfig:
|
def create_model_config(self) -> ModelConfig:
|
||||||
|
|||||||
Reference in New Issue
Block a user