plaguss
/

Qwen2.5-0.5B-Math-Shepherd-PRM-token-0.1

Token Classification

Generated from Trainer

stepwise-reward-trainer

text-generation-inference

Model card Files Files and versions Community

Qwen2.5-0.5B-Math-Shepherd-PRM-token-0.1

1 contributor

History: 4 commits

plaguss's picture

Training in progress, step 1500

01843ef verified 4 months ago

.gitattributes

1.57 kB

Training in progress, step 500 4 months ago
added_tokens.json

605 Bytes

Training in progress, step 500 4 months ago
config.json

787 Bytes

Training in progress, step 500 4 months ago
merges.txt

1.67 MB

Training in progress, step 500 4 months ago
model.safetensors

988 MB
LFS

Training in progress, step 1500 4 months ago
special_tokens_map.json

616 Bytes

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

11.4 MB
LFS

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

7.23 kB

Training in progress, step 500 4 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 4 months ago
vocab.json

2.78 MB

Training in progress, step 500 4 months ago