"examples/vscode:/vscode.git/clone" did not exist on "813328682e2a6822ff8d0fde30a7ed9012449daf"
Unverified Commit 3b5fa14f authored by Timothé Pearce's avatar Timothé Pearce Committed by GitHub
Browse files

Fix `wandb` integration with `SetFit` model (#30021)



Fix W&B integration with SetFit model
Co-authored-by: default avatarPEARCE Timothe <timothe_pearce@ext.connect-tech.sncf>
parent f4d18944
......@@ -753,7 +753,7 @@ class WandbCallback(TrainerCallback):
combined_dict = {**args.to_dict()}
if hasattr(model, "config") and model.config is not None:
model_config = model.config.to_dict()
model_config = model.config if isinstance(model.config, dict) else model.config.to_dict()
combined_dict = {**model_config, **combined_dict}
if hasattr(model, "peft_config") and model.peft_config is not None:
peft_config = model.peft_config
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment