mirror of
https://github.com/hanzo-ml/katib.git
synced 2026-09-24 00:31:27 +00:00
[SDK] fix issue with serialization of LoraConfig (#2567)
Signed-off-by: Zackeus Bengtsson <[email protected]>
This commit is contained in:
committed by
GitHub
parent
55550f8df5
commit
2adf3a92c2
@@ -188,8 +188,9 @@ def get_trial_substitutions_from_trainer(
|
||||
if isinstance(parameters, TrainingArguments):
|
||||
parameters_dict = parameters.to_dict()
|
||||
else:
|
||||
parameters_dict = parameters.__dict__
|
||||
|
||||
parameters_dict = (
|
||||
parameters.to_dict() if hasattr(parameters, "to_dict") else vars(parameters)
|
||||
)
|
||||
for p_name, p_value in parameters_dict.items():
|
||||
if not hasattr(parameters, p_name):
|
||||
logger.warning(f"Training parameter {p_name} is not supported.")
|
||||
@@ -218,7 +219,11 @@ def get_trial_substitutions_from_trainer(
|
||||
if isinstance(parameters, TrainingArguments):
|
||||
parameters = json.dumps(parameters.to_dict())
|
||||
else:
|
||||
parameters = json.dumps(parameters.__dict__, cls=SetEncoder)
|
||||
parameters = (
|
||||
json.dumps(parameters.to_dict(), cls=SetEncoder)
|
||||
if hasattr(parameters, "to_dict")
|
||||
else json.dumps(vars(parameters), cls=SetEncoder)
|
||||
)
|
||||
|
||||
return parameters
|
||||
|
||||
|
||||
Reference in New Issue
Block a user