From 09c0673ded65578269926ff4d5547456be7c7f5a Mon Sep 17 00:00:00 2001
From: litagin02 <litagin02@gmail.com>
Date: Sat, 6 Jan 2024 15:07:44 +0900
Subject: [PATCH] Reduce bert_gen numprocess to 2 (often freezes in colab)

---
 default_config.yml | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/default_config.yml b/default_config.yml
index ab5bebc..99d7601 100644
--- a/default_config.yml
+++ b/default_config.yml
@@ -22,7 +22,7 @@ preprocess_text:
 
 bert_gen:
   config_path: "config.json"
-  num_processes: 4
+  num_processes: 2
   device: "cuda"
   use_multi_device: false