Update sample.py

sample.py

@@ -3,7 +3,7 @@ import torch.nn.functional as F
 
 from utils import limit_past, kl, entropy
 
-def sample(model, enc, length, context, temperature=1.0, device='cuda', topk=-1):
+def sample(model, enc, length, context, temperature=1.0, device='cpu', topk=-1):
     assert length > 0
 
     context = torch.tensor(context[-1022:], device=device, dtype=torch.long)