From d8abe323a8c7484f29693ddb5a3fa9ca202f119b Mon Sep 17 00:00:00 2001 From: Disk-MTH Date: Fri, 5 Dec 2025 17:59:40 +0100 Subject: [PATCH] Add depth_anything model config for export --- optimum/exporters/onnx/model_configs.py | 5 +++++ 1 file changed, 5 insertions(+) diff --git a/optimum/exporters/onnx/model_configs.py b/optimum/exporters/onnx/model_configs.py index c4ac1c39..03df2276 100644 --- a/optimum/exporters/onnx/model_configs.py +++ b/optimum/exporters/onnx/model_configs.py @@ -1037,6 +1037,11 @@ class GlpnOnnxConfig(ViTOnnxConfig): pass +@register_tasks_manager_onnx("depth_anything", *["feature-extraction", "depth-estimation"]) +class DepthAnythingOnnxConfig(DptOnnxConfig): + pass + + @register_tasks_manager_onnx("poolformer", *["feature-extraction", "image-classification"]) class PoolFormerOnnxConfig(ViTOnnxConfig): NORMALIZED_CONFIG_CLASS = NormalizedVisionConfig