8 contributors

History: 31 commits

versae

New Flax model

300e533 over 3 years ago

configs
Adding base config and organizing configs over 3 years ago
mc4
Fixes and defaults over 3 years ago
.gitattributes

736 Bytes

Update .gitattributes over 3 years ago
.gitignore

1.84 kB

Initial test with BETO's corpus over 3 years ago
README.md

1.84 kB

Adding checkpointing, wandb, and new mlm script over 3 years ago
config.json

618 Bytes

Adding config and models for the hub widget over 3 years ago
config.py

247 Bytes

Scripts for perplexity sampling and fixes over 3 years ago
convert.py

469 Bytes

Adding missing import over 3 years ago
flax_model.msgpack

250 MB
LFS

New Flax model over 3 years ago
get_embeddings_and_perplexity.py

1.53 kB

Add script to generate dataset of embeddings and perplexities. Add script to generate t-SNE plot for embedding and perplexity visualization. over 3 years ago
perplexity.py

751 Bytes

Adding checkpointing, wandb, and new mlm script over 3 years ago
pytorch_model.bin

499 MB
LFS

Base model at 105k steps over 3 years ago
run.sh

883 Bytes

Adding base config and organizing configs over 3 years ago
run_mlm_flax.py

30 kB

Adding sampling to mc4 over 3 years ago
run_mlm_flax_stream.py

28.1 kB

Fixes and defaults over 3 years ago
run_stream.sh

924 Bytes

Adding base config and organizing configs over 3 years ago
tokenizer.json

1.47 MB

New tokenizer over 3 years ago
tokens.py

649 Bytes

Scripts for perplexity sampling and fixes over 3 years ago
tokens.py.orig

899 Bytes

Adjust batch size for extrating tokens over 3 years ago
tsne_plot.py

3.02 kB

Remove unused imports over 3 years ago