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

Upload convert_flax_to_pytorch.py

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