Upload better_transformer.py

better_transformer.py

@@ -361,7 +361,7 @@ def generate(model, tokenizer, device, method=None, k=None,
 
     if cond != "":
 
-        cond_tokens = tokenizer(cond).input_ids.to(device) ## HERE???
+        cond_tokens = tokenizer(cond).input_ids
 
         gen_tokens = model.generate(torch.tensor(cond_tokens).unsqueeze(0).long().to(device),
                                     method=method, k=k, p_nucleus=p_nucleus, temp=temp,