update to v0.2
b98a8b6
verified
-
onnx
Upload ONNX weights (#1)
-
runs
End of training
-
1.64 kB
Upload ONNX weights (#1)
-
3.35 kB
Update code example in readme to use `device` variable instead of hard coded `cuda` (#9)
-
783 Bytes
End of training
-
738 Bytes
End of training
-
587 Bytes
End of training
-
132 Bytes
Model save
-
466 kB
Model save
-
655 Bytes
Model save
-
2.1 MB
Model save
-
3.59 kB
Model save
-
231 Bytes
Model save
-
4.66 kB
Model save
training_args.bin
Detected Pickle imports (13)
- "transformers.trainer_utils.HubStrategy",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.integrations.deepspeed.HfDeepSpeedConfig",
- "accelerate.utils.dataclasses.DeepSpeedPlugin",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "transformers.trainer_utils.IntervalStrategy",
- "accelerate.state.PartialState",
- "torch.bfloat16",
- "alignment.configs.DPOConfig",
- "transformers.trainer_utils.SchedulerType",
- "torch.device",
- "transformers.integrations.deepspeed.HfTrainerDeepSpeedConfig"
How to fix it?
6.52 kB
Model save
-
801 kB
Model save