JayHyeon
/

pythia-2.8b-DPO_1e-6_1.0vpo_constant-1ep

Text Generation

Generated from Trainer

text-generation-inference

Model card Files Files and versions Community

pythia-2.8b-DPO_1e-6_1.0vpo_constant-1ep

5.55 GB

1 contributor

History: 2 commits

JayHyeon's picture

Training in progress, step 971

be831c3 verified 3 months ago

.gitattributes

1.52 kB

initial commit 3 months ago
chat_template.jinja

167 Bytes

Training in progress, step 971 3 months ago
config.json

761 Bytes

Training in progress, step 971 3 months ago
model-00001-of-00002.safetensors

4.98 GB
LFS

Training in progress, step 971 3 months ago
model-00002-of-00002.safetensors

572 MB
LFS

Training in progress, step 971 3 months ago
model.safetensors.index.json

34 kB

Training in progress, step 971 3 months ago
special_tokens_map.json

473 Bytes

Training in progress, step 971 3 months ago
tokenizer.json

3.56 MB

Training in progress, step 971 3 months ago
tokenizer_config.json

4.88 kB

Training in progress, step 971 3 months ago
training_args.bin
Detected Pickle imports (15)
- "trl.trainer.dpo_config.DPOConfig",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "accelerate.utils.dataclasses.DeepSpeedPlugin",
- "torch.bfloat16",
- "torch.device",
- "transformers.trainer_utils.IntervalStrategy",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_utils.HubStrategy",
- "transformers.integrations.deepspeed.HfTrainerDeepSpeedConfig",
- "transformers.integrations.deepspeed.HfDeepSpeedConfig",
- "trl.trainer.dpo_config.FDivergenceType",
- "transformers.trainer_utils.SaveStrategy",
- "transformers.trainer_utils.SchedulerType",
- "accelerate.state.PartialState"
How to fix it?
8.21 kB
LFS

Training in progress, step 971 3 months ago