Spaces:
Runtime error
Runtime error
akshaytrikha
commited on
Commit
β’
ba07099
1
Parent(s):
505265f
rename model to gpt2-124M
Browse files
app.py
CHANGED
@@ -5,7 +5,7 @@ import torch
|
|
5 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
6 |
|
7 |
# load pretrained + finetuned GPT2
|
8 |
-
model = GPT2LMHeadModel.from_pretrained("./model")
|
9 |
model = model.to(device)
|
10 |
|
11 |
# create tokenizer
|
|
|
5 |
device = 'cuda' if torch.cuda.is_available() else 'cpu'
|
6 |
|
7 |
# load pretrained + finetuned GPT2
|
8 |
+
model = GPT2LMHeadModel.from_pretrained("./model/gpt2-124M")
|
9 |
model = model.to(device)
|
10 |
|
11 |
# create tokenizer
|
huggingface_trump.ipynb
CHANGED
The diff for this file is too large to render.
See raw diff
|
|
model/.DS_Store
ADDED
Binary file (6.15 kB). View file
|
|
model/{config.json β gpt2-124M/config.json}
RENAMED
File without changes
|
model/{pytorch_model.bin β gpt2-124M/pytorch_model.bin}
RENAMED
File without changes
|
model/{training_args.bin β gpt2-124M/training_args.bin}
RENAMED
File without changes
|