plaguss
/

Mistral-7B-v0.1-Math-Shepherd-PRM-0.1

Token Classification

Generated from Trainer

stepwise-reward-trainer

text-generation-inference

Model card Files Files and versions Community

Mistral-7B-v0.1-Math-Shepherd-PRM-0.1

Ctrl+K

Ctrl+K

1 contributor

History: 5 commits

plaguss's picture

Training in progress, step 1650

c64258f verified 5 months ago

.gitattributes

1.52 kB

initial commit 5 months ago
config.json

711 Bytes

Training in progress, step 500 5 months ago
model-00001-of-00003.safetensors

4.94 GB
LFS

Training in progress, step 1650 5 months ago
model-00002-of-00003.safetensors

5 GB
LFS

Training in progress, step 1650 5 months ago
model-00003-of-00003.safetensors

4.28 GB
LFS

Training in progress, step 1650 5 months ago
model.safetensors.index.json

24 kB

Training in progress, step 500 5 months ago
special_tokens_map.json

557 Bytes

Training in progress, step 500 5 months ago
tokenizer.json

3.51 MB

Training in progress, step 500 5 months ago
tokenizer_config.json

1.64 kB

Training in progress, step 500 5 months ago
training_args.bin
Detected Pickle imports (13)
- "trl.trainer.stepwise_reward_config.StepwiseRewardConfig",
- "transformers.trainer_utils.IntervalStrategy",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.trainer_pt_utils.AcceleratorConfig",
- "transformers.trainer_utils.SchedulerType",
- "transformers.training_args.OptimizerNames",
- "transformers.integrations.deepspeed.HfDeepSpeedConfig",
- "accelerate.state.PartialState",
- "torch.device",
- "accelerate.utils.dataclasses.DeepSpeedPlugin",
- "torch.bfloat16",
- "transformers.trainer_utils.HubStrategy",
- "transformers.integrations.deepspeed.HfTrainerDeepSpeedConfig"
How to fix it?
6.71 kB
LFS

Training in progress, step 500 5 months ago