Skip to content

Commit 31aec62

Browse files
committed
Make parameters config optional in RawPyTorchMergeConfig.
1 parent d4b4b6c commit 31aec62

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

mergekit/scripts/merge_raw_pytorch.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -28,8 +28,8 @@ class InputModelDefinition(BaseModel, frozen=True):
2828

2929
class RawPyTorchMergeConfig(BaseModel, frozen=True):
3030
merge_method: str
31-
parameters: Optional[Dict[str, ParameterSetting]]
3231
models: List[InputModelDefinition]
32+
parameters: Optional[Dict[str, ParameterSetting]] = None
3333
dtype: Optional[str] = None
3434
base_model: Optional[str] = None
3535

@@ -169,7 +169,7 @@ def construct_param_dicts(
169169
):
170170
global_params = {}
171171
for param_def in merge_method.parameters():
172-
if param_def.name in config.parameters:
172+
if config.parameters and param_def.name in config.parameters:
173173
value = evaluate_setting(tensor_name, config.parameters[param_def.name])
174174
if value is not None:
175175
global_params[param_def.name] = value

0 commit comments

Comments
 (0)