Commit
•
0c7ede3
1
Parent(s):
5798f9c
Update README.md
Browse files
README.md
CHANGED
@@ -31,8 +31,8 @@ import torch
|
|
31 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
32 |
print(device)
|
33 |
|
34 |
-
model = GPT2LMHeadModel.from_pretrained("erythropygia/gpt2-base
|
35 |
-
tokenizer = GPT2TokenizerFast.from_pretrained("erythropygia/gpt2-base
|
36 |
tokenizer.pad_token = tokenizer.eos_token
|
37 |
|
38 |
def generate_output(text):
|
|
|
31 |
device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
|
32 |
print(device)
|
33 |
|
34 |
+
model = GPT2LMHeadModel.from_pretrained("erythropygia/gpt2-turkish-base").to(device)
|
35 |
+
tokenizer = GPT2TokenizerFast.from_pretrained("erythropygia/gpt2-turkish-base")
|
36 |
tokenizer.pad_token = tokenizer.eos_token
|
37 |
|
38 |
def generate_output(text):
|