philschmid HF staff commited on
Commit
8b41d3d
1 Parent(s): 8e2a3c8

Upload convert_pytorch_to_flax.py

Browse files
Files changed (1) hide show
  1. convert_pytorch_to_flax.py +3 -0
convert_pytorch_to_flax.py ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ from transformers import FlaxAutoModelForSequenceClassification
2
+ model = FlaxAutoModelForSequenceClassification.from_pretrained("./", from_pt=True)
3
+ model.save_pretrained("./")