Update cpu_count

This commit is contained in:
Andrei Betlen
2023-05-07 03:03:57 -04:00
parent 1a00e452ea
commit 5a3413eee3

View File

@@ -1,8 +1,8 @@
import os
import json import json
import multiprocessing
from threading import Lock from threading import Lock
from typing import List, Optional, Union, Iterator, Dict from typing import List, Optional, Union, Iterator, Dict
from typing_extensions import TypedDict, Literal, Annotated from typing_extensions import TypedDict, Literal
import llama_cpp import llama_cpp
@@ -21,7 +21,7 @@ class Settings(BaseSettings):
default=512, ge=1, description="The batch size to use per eval." default=512, ge=1, description="The batch size to use per eval."
) )
n_threads: int = Field( n_threads: int = Field(
default=max((os.cpu_count() or 2) // 2, 1), default=max(multiprocessing.cpu_count() // 2, 1),
ge=1, ge=1,
description="The number of threads to use.", description="The number of threads to use.",
) )