File size: 147 Bytes
56d26ba |
1 2 3 4 5 6 |
from transformers import AutoModelForCausalLM
pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True)
pt_model.save_pretrained(".")
|
56d26ba |
1 2 3 4 5 6 |
from transformers import AutoModelForCausalLM
pt_model = AutoModelForCausalLM.from_pretrained('.', from_flax=True)
pt_model.save_pretrained(".")
|