adjust onnx
This commit is contained in:
@@ -509,10 +509,10 @@ module Templates
|
|||||||
def model
|
def model
|
||||||
@model ||= OnnxRuntime::Model.new(
|
@model ||= OnnxRuntime::Model.new(
|
||||||
MODEL_PATH.to_s,
|
MODEL_PATH.to_s,
|
||||||
inter_op_num_threads: CPU_THREADS,
|
inter_op_num_threads: 1,
|
||||||
intra_op_num_threads: CPU_THREADS,
|
intra_op_num_threads: CPU_THREADS,
|
||||||
enable_mem_pattern: false,
|
enable_mem_pattern: false,
|
||||||
enable_cpu_mem_arena: false,
|
enable_cpu_mem_arena: Docuseal.multitenant? || Rails.env.development?,
|
||||||
providers: ['CPUExecutionProvider']
|
providers: ['CPUExecutionProvider']
|
||||||
)
|
)
|
||||||
end
|
end
|
||||||
|
|||||||
Reference in New Issue
Block a user