Hugging Face
Models
Datasets
Spaces
Posts
Docs
Solutions
Pricing
Log In
Sign Up
imthanhlv
/
vigpt2medium
like
1
Text Generation
Transformers
PyTorch
JAX
gpt2
Inference Endpoints
text-generation-inference
Model card
Files
Files and versions
Community
15
Train
Deploy
Use this model
refs/pr/12
vigpt2medium
/
flax_to_pt.py
imthanhlv
add torch model
56d26ba
over 2 years ago
raw
history
blame
No virus
147 Bytes
from
transformers
import
AutoModelForCausalLM
pt_model = AutoModelForCausalLM.from_pretrained(
'.'
, from_flax=
True
)
pt_model.save_pretrained(
"."
)