File size: 147 Bytes
1f656b0
 
 
 
 
1
2
3
4
5
6
from transformers import AutoModelForCausalLM


pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True)
pt_model.save_pretrained(".")