1f656b0
1
2
3
4
5
6
from transformers import AutoModelForCausalLM pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True) pt_model.save_pretrained(".")