adjust onnx

pull/572/head
Pete Matsyburka 4 weeks ago
parent 4375459856
commit 40f8093e00

@ -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

Loading…
Cancel
Save