Update generate.py

generate.py

@@ -63,8 +63,7 @@ class GenerateRunner():
         # 加载模型
         file_name = os.path.join(opt.model_path, f'model_{opt.epoch}.pt')
         if opt.model_choice == 'transformer':
-            # self.model = EncoderDecoder.load_from_file(file_name)
-            self.model = EncoderDecoder.load_from_file(file_name, map_location=torch.device('cpu'))
+            self.model = EncoderDecoder.load_from_file(file_name)
             self.model.to(self.device)
             self.model.eval()
         elif opt.model_choice == 'seq2seq':