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

Upload convert_pytorch_to_tensorflow.py

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