more graceful timeout
This commit is contained in:
@@ -125,7 +125,7 @@ def test_context_size(model_name: str, num_ctx: int) -> Optional[Dict]:
|
|||||||
|
|
||||||
# Send request with longer timeout for large contexts
|
# Send request with longer timeout for large contexts
|
||||||
# Large contexts can take time to allocate
|
# Large contexts can take time to allocate
|
||||||
timeout = 60 if num_ctx > 100000 else 30
|
timeout = 120 if num_ctx > 100000 else 60
|
||||||
|
|
||||||
with urllib.request.urlopen(req, timeout=timeout) as response:
|
with urllib.request.urlopen(req, timeout=timeout) as response:
|
||||||
response_data = response.read().decode()
|
response_data = response.read().decode()
|
||||||
|
|||||||
Reference in New Issue
Block a user