.

app.py

@@ -174,7 +174,7 @@ block_size = min(block_size, tokenizer_decoder.max_len_single_sentence)
 
 # Load full model
 output_full_dir = '/home/ryn_mote/Misc/generative_recommender/text_space/' 
-checkpoint = torch.load(os.path.join(model_path, 'training.bin'))
+checkpoint = torch.load(os.path.join(model_path, 'training.bin'), map_location=torch.device(DEVICE))
 
 # Chunyuan: Add Padding token to GPT2
 special_tokens_dict = {'pad_token': '<PAD>', 'bos_token': '<BOS>', 'eos_token': '<EOS>'}