4 contributors

History: 12 commits

hl-tburns

Update code example in readme to use `device` variable instead of hard coded `cuda`

8d04040 verified 5 months ago

onnx
Upload ONNX weights (#1) 5 months ago
runs
End of training 5 months ago
.gitattributes

1.52 kB

initial commit 5 months ago
README.md

3.31 kB

Update code example in readme to use `device` variable instead of hard coded `cuda` 5 months ago
all_results.json

787 Bytes

End of training 5 months ago
config.json

735 Bytes

End of training 5 months ago
eval_results.json

592 Bytes

End of training 5 months ago
generation_config.json

132 Bytes

Model save 5 months ago
merges.txt

466 kB

Model save 5 months ago
model.safetensors

269 MB
LFS

Model save 5 months ago
special_tokens_map.json

655 Bytes

Model save 5 months ago
tokenizer.json

2.1 MB

Model save 5 months ago
tokenizer_config.json

3.59 kB

Model save 5 months ago
train_results.json

230 Bytes

Model save 5 months ago
trainer_state.json

4.65 kB

Model save 5 months ago
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
LFS

Model save 5 months ago
vocab.json

801 kB

Model save 5 months ago

Detected Pickle imports (13)