bhadresh-savani commited on
Commit
0c24d02
1 Parent(s): b0d0944

Added pytorch model

Browse files
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "/home/patrick/t5/byt5-base",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
@@ -23,6 +23,7 @@
23
  "relative_attention_num_buckets": 32,
24
  "tie_word_embeddings": false,
25
  "tokenizer_class": "ByT5Tokenizer",
 
26
  "transformers_version": "4.9.0.dev0",
27
  "use_cache": true,
28
  "vocab_size": 384
1
  {
2
+ "_name_or_path": "./",
3
  "architectures": [
4
  "T5ForConditionalGeneration"
5
  ],
23
  "relative_attention_num_buckets": 32,
24
  "tie_word_embeddings": false,
25
  "tokenizer_class": "ByT5Tokenizer",
26
+ "torch_dtype": "float32",
27
  "transformers_version": "4.9.0.dev0",
28
  "use_cache": true,
29
  "vocab_size": 384
convert_to_pytorch.py ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ from transformers import AutoModelForSeq2SeqLM
2
+ model = AutoModelForSeq2SeqLM.from_pretrained("./", from_flax=True)
3
+ model.save_pretrained("./")
convert_to_tensorflow.py ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ from transformers import TFAutoModelForSeq2SeqLM
2
+ model = TFAutoModelForSeq2SeqLM.from_pretrained("./", from_pt=True)
3
+ model.save_pretrained("./")
pytorch_model.bin ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c4d23666ccdaf3dc0983b49747bc48ac4c8eba8344e29c42ec0f8e1258c75932
3
+ size 2326723273
tf_model.h5 ADDED
@@ -0,0 +1,3 @@
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:cb50cf814adc9647925e63cf9e64a5b2b484a5464307317631eea0e5959413e1
3
+ size 2327126272