JuncaiL
/

llama-265m

Text Generation

Model card Files Files and versions Community

1 contributor

History: 4 commits

JuncaiL's picture

fix state_dict loading in MoE model

3240d88 verified 8 months ago

.gitattributes

1.52 kB

initial commit 8 months ago
README.md

1 Bytes

upload llama-265m model checkpoint 8 months ago
config.json

1.39 kB

fix state_dict loading in MoE model 8 months ago
configuration_llama_moe.py

4.41 kB

upload llama-265m model checkpoint 8 months ago
generation_config.json

132 Bytes

upload llama-265m model checkpoint 8 months ago
modeling_llama_moe_hf.py

66.7 kB

fix state_dict loading in MoE model 8 months ago
pytorch_model.bin
Detected Pickle imports (3)
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage",
- "collections.OrderedDict"
What is a pickle import?
1.06 GB
LFS

upload llama-265m model checkpoint 8 months ago
special_tokens_map.json

411 Bytes

upload llama-265m model checkpoint 8 months ago
tokenizer.model

500 kB
LFS

upload llama-265m model checkpoint 8 months ago
tokenizer_config.json

720 Bytes

upload llama-265m model checkpoint 8 months ago
trainer_state.json

72.6 kB

upload llama-265m model checkpoint 8 months ago
training_args.bin
Detected Pickle imports (8)
- "torch.device",
- "transformers.trainer_utils.IntervalStrategy",
- "transformers.trainer_utils.HubStrategy",
- "accelerate.state.PartialState",
- "accelerate.utils.dataclasses.DistributedType",
- "transformers.training_args.OptimizerNames",
- "transformers.trainer_utils.SchedulerType",
- "transformers.training_args.TrainingArguments"
How to fix it?
3.95 kB
LFS

upload llama-265m model checkpoint 8 months ago