Deeokay
/

gpt2-javis-stks

Text Generation

text-generation-inference

Inference Endpoints

Model card Files Files and versions Community

Deeokay commited on Jun 15

Commit

770a148

•

1 Parent(s): c01c297

Update README.md

Files changed (1) hide show

README.md +1 -1

README.md CHANGED Viewed

@@ -51,7 +51,7 @@ from transformers import GPT2LMHeadModel, GPT2Tokenizer
 models_folder = "Deeokay/gpt2-javis-stks"
 # if you know your device, you can just set "device = 'mps'"
-device = torch.device("cuda" if torch.cuda.is_available90 else "cpu")
 model = GPT2LMHeadModel.from_pretrained(models_folder)
 tokenizer = GPT2Tokenizer.from_pretrained(models_folder)

 models_folder = "Deeokay/gpt2-javis-stks"
 # if you know your device, you can just set "device = 'mps'"
+device = torch.device("cuda" if torch.cuda.is_available() else "cpu")
 model = GPT2LMHeadModel.from_pretrained(models_folder)
 tokenizer = GPT2Tokenizer.from_pretrained(models_folder)