koshkosh commited on
Commit
f28517c
·
verified ·
1 Parent(s): 9c16faf

Upload python_gpt2.py

Browse files
Files changed (1) hide show
  1. python_gpt2.py +12 -0
python_gpt2.py ADDED
@@ -0,0 +1,12 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import GPT2LMHeadModel, GPT2Tokenizer
2
+
3
+ # Model name to download
4
+ model_name = "gpt2"
5
+
6
+ # Download the GPT-2 model and tokenizer
7
+ model = GPT2LMHeadModel.from_pretrained(model_name)
8
+ tokenizer = GPT2Tokenizer.from_pretrained(model_name)
9
+
10
+ # Save the model and tokenizer files locally
11
+ model.save_pretrained("./gpt2_model")
12
+ tokenizer.save_pretrained("./gpt2_tokenizer")