File tree 2 files changed +2
-4
lines changed
2 files changed +2
-4
lines changed Original file line number Diff line number Diff line change @@ -262,9 +262,7 @@ def __init__(
262
262
263
263
self .n_batch = min (n_ctx , n_batch ) # ???
264
264
self .n_threads = n_threads or max (multiprocessing .cpu_count () // 2 , 1 )
265
- self .n_threads_batch = n_threads_batch or max (
266
- multiprocessing .cpu_count () // 2 , 1
267
- )
265
+ self .n_threads_batch = n_threads_batch or multiprocessing .cpu_count ()
268
266
269
267
# Context Params
270
268
self .context_params = llama_cpp .llama_context_default_params ()
Original file line number Diff line number Diff line change @@ -70,7 +70,7 @@ class ModelSettings(BaseSettings):
70
70
description = "The number of threads to use." ,
71
71
)
72
72
n_threads_batch : int = Field (
73
- default = max (multiprocessing .cpu_count () // 2 , 1 ),
73
+ default = max (multiprocessing .cpu_count (), 1 ),
74
74
ge = 0 ,
75
75
description = "The number of threads to use when batch processing." ,
76
76
)
You can’t perform that action at this time.
0 commit comments