Skip to content

Instantly share code, notes, and snippets.

@epwalsh

epwalsh/step2.py

Created August 13, 2020 18:19
Show Gist options
  • Save epwalsh/bcc92279942e9a8db0e545170f11a134 to your computer and use it in GitHub Desktop.
Save epwalsh/bcc92279942e9a8db0e545170f11a134 to your computer and use it in GitHub Desktop.
Download ZIP
How to upload transformer weights and tokenizers from AllenNLP models to HuggingFace's model hub: step 2
Raw
step2.py
transformer_dir = "~/my-cool-transformer"
transformer_embedder.transformer_model.save_pretrained(transformer_dir)
tokenizer.tokenizer.save_pretrained(transformer_dir)
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment