schoginitoys commited on
Commit
5548e34
·
verified ·
1 Parent(s): 5ebdf85

Create download_and_save_gpt2.py

Browse files
Files changed (1) hide show
  1. download_and_save_gpt2.py +9 -0
download_and_save_gpt2.py ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ from transformers import GPT2Model, GPT2TokenizerFast
2
+
3
+ model = GPT2Model.from_pretrained("gpt2")
4
+ tokenizer = GPT2TokenizerFast.from_pretrained("gpt2")
5
+
6
+ model.save_pretrained("models")
7
+ tokenizer.save_pretrained("models")
8
+
9
+ print("✅ Downloaded and saved GPT-2 to ./models")