reduce num worker

infer_concat.py

@@ -104,6 +104,6 @@ def infer_2_hier(model, data_loader, device, tokenizer):
 
 def vit5_infer(data):
 	dataset = Dataset4Summarization(data, tokenizer)
-	data_loader = torch.utils.data.DataLoader(dataset, batch_size=1, num_workers=2)
+	data_loader = torch.utils.data.DataLoader(dataset, batch_size=1, num_workers=1)
 	result = infer_2_hier(model, data_loader, device, tokenizer)
 	return result