浏览代码

Increase timeout

Max Ryabinin 3 年之前
父节点
当前提交
66e387d23c
共有 1 个文件被更改,包括 1 次插入0 次删除
  1. 1 0
      hivemind/moe/server/__init__.py

+ 1 - 0
hivemind/moe/server/__init__.py

@@ -303,6 +303,7 @@ class Server(threading.Thread):
                     metadata_expiration=metadata_expiration,
                     averaging_timeout=averaging_timeout,
                     reuse_grad_buffers=reuse_grad_buffers,
+                    request_timeout=10,
                     verbose=True,
                     start=True,
                 )