Shiv-22
/

diffmamba-checkpoints

Text Generation

masked-diffusion

diffusion-language-model

state-space-model

pytorch-lightning

Model card Files Files and versions

diffmamba-checkpoints / runs /gen_stage2 /checkpoints

Ctrl+K

Ctrl+K

1 contributor

History: 6 commits

Shiv-22's picture

prune 153 redundant checkpoints

cac3ae9 verified about 2 months ago

best.ckpt
Detected Pickle imports (17)
- "torch.LongStorage",
- "omegaconf.nodes.AnyNode",
- "tokenizers.models.Model",
- "torch._utils._rebuild_tensor_v2",
- "torch.FloatStorage",
- "transformers.models.gpt2.tokenization_gpt2.GPT2Tokenizer",
- "tokenizers.AddedToken",
- "typing.Any",
- "__builtin__.dict",
- "omegaconf.base.ContainerMetadata",
- "omegaconf.base.Metadata",
- "_codecs.encode",
- "torch.DoubleStorage",
- "collections.defaultdict",
- "tokenizers.Tokenizer",
- "collections.OrderedDict",
- "omegaconf.dictconfig.DictConfig"
How to fix it?
2.11 GB
xet

Upload runs/gen_stage2/checkpoints/best.ckpt with huggingface_hub about 2 months ago
last.ckpt
Detected Pickle imports (17)
- "tokenizers.models.Model",
- "torch.FloatStorage",
- "collections.OrderedDict",
- "omegaconf.base.ContainerMetadata",
- "tokenizers.Tokenizer",
- "typing.Any",
- "omegaconf.base.Metadata",
- "transformers.models.gpt2.tokenization_gpt2.GPT2Tokenizer",
- "torch.DoubleStorage",
- "torch._utils._rebuild_tensor_v2",
- "torch.LongStorage",
- "collections.defaultdict",
- "__builtin__.dict",
- "omegaconf.dictconfig.DictConfig",
- "_codecs.encode",
- "tokenizers.AddedToken",
- "omegaconf.nodes.AnyNode"
How to fix it?
2.11 GB
xet

Upload runs/gen_stage2/checkpoints/last.ckpt with huggingface_hub about 2 months ago