imthanhlv commited on
Commit
1f656b0
1 Parent(s): b80a01b

Saving weights and logs of step 37500

Browse files
Files changed (2) hide show
  1. flax_model.msgpack +1 -1
  2. flax_to_pt.py +5 -0
flax_model.msgpack CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:f39fc1a6debdcb6bb875f04299e35060e9514f0616e1a7d041ac6c01ab3f9aa0
3
  size 497764120
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:c448025ff2329ac30909b08df717b71777d6b68618aebd5d5167911313091df0
3
  size 497764120
flax_to_pt.py ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
1
+ from transformers import AutoModelForCausalLM
2
+
3
+
4
+ pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True)
5
+ pt_model.save_pretrained(".")