[model] add QwQ 32b (#7179)

Former-commit-id: 8897e48b8cd55407812453ddd4ff98ac7bdc4e91
This commit is contained in:
hoshi-hiyouga
2025-03-06 11:58:36 +08:00
committed by GitHub
parent 25bb9f5ad9
commit 9f16c50155
2 changed files with 6 additions and 3 deletions

View File

@@ -600,15 +600,14 @@ def get_swanlab_callback(finetuning_args: "FinetuningArguments") -> "TrainerCall
return
super().setup(args, state, model, **kwargs)
try:
if hasattr(self, "_swanlab"):
swanlab_public_config = self._swanlab.get_run().public.json()
else: # swanlab <= 0.4.9
swanlab_public_config = self._experiment.get_run().public.json()
except Exception as e:
except Exception:
swanlab_public_config = {}
with open(os.path.join(args.output_dir, SWANLAB_CONFIG), "w") as f:
f.write(json.dumps(swanlab_public_config, indent=2))