Spaces:

IkIzma
/

Transformer_Homework

Runtime error

IkIzma commited on Apr 16, 2023

Commit

361d939

1 Parent(s): fe1596b

Update Generate_text.py

Files changed (1) hide show

Generate_text.py CHANGED Viewed

@@ -7,7 +7,7 @@ torch.manual_seed(17)
 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 def load_tokenizer_and_model(model_name_or_path, device):
-    return GPT2Tokenizer.from_pretrained(model_name_or_path), GPT2LMHeadModel.from_pretrained(model_name_or_path).cuda()
 def generate(
     model, tok, text, device,

 from transformers import GPT2LMHeadModel, GPT2Tokenizer
 def load_tokenizer_and_model(model_name_or_path, device):
+    return GPT2Tokenizer.from_pretrained(model_name_or_path), GPT2LMHeadModel.from_pretrained(model_name_or_path).to(device)
 def generate(
     model, tok, text, device,