PratikGautam commited on
Commit
805db4c
·
verified ·
1 Parent(s): ca06dfe

Create model/pytorch_model.bin

Browse files
Files changed (1) hide show
  1. model/pytorch_model.bin +5 -0
model/pytorch_model.bin ADDED
@@ -0,0 +1,5 @@
 
 
 
 
 
 
1
+ from transformers import AutoModelForCausalLM, AutoTokenizer
2
+
3
+ # Load GPT-2 small
4
+ model = AutoModelForCausalLM.from_pretrained("gpt2")
5
+ tokenizer = AutoTokenizer.from_pretrained("gpt2")