[model] add QwQ 32b (#7179)
Former-commit-id: 8897e48b8cd55407812453ddd4ff98ac7bdc4e91
This commit is contained in:
@@ -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))
|
||||
|
||||
|
||||
Reference in New Issue
Block a user