diff --git a/examples/recipes/README.md b/examples/recipes/README.md index caaa2c15f..2c7f46e56 100644 --- a/examples/recipes/README.md +++ b/examples/recipes/README.md @@ -16,6 +16,7 @@ Each *(model, task)* includes: | Model | Task | |---|---| +| apple/DepthPro-hf | depth-estimation | | BAAI/bge-large-en-v1.5 | sentence-similarity | | cardiffnlp/twitter-roberta-base-sentiment-latest | text-classification | | deepset/roberta-base-squad2 | question-answering | diff --git a/examples/recipes/apple_DepthPro-hf/depth-estimation_fp16_config.json b/examples/recipes/apple_DepthPro-hf/depth-estimation_fp16_config.json new file mode 100644 index 000000000..b253572d9 --- /dev/null +++ b/examples/recipes/apple_DepthPro-hf/depth-estimation_fp16_config.json @@ -0,0 +1,45 @@ +{ + "export": { + "opset_version": 17, + "batch_size": 1, + "export_params": true, + "do_constant_folding": true, + "verbose": false, + "dynamo": false, + "enable_hierarchy_tags": true, + "clean_onnx": false, + "hierarchy_tag_format": "full", + "input_tensors": [ + { + "name": "pixel_values", + "dtype": "float32", + "shape": [ + 1, + 3, + 1536, + 1536 + ], + "value_range": [ + 0, + 1 + ] + } + ], + "output_tensors": [ + { + "name": "predicted_depth" + }, + { + "name": "field_of_view" + } + ] + }, + "optim": {}, + "quant": null, + "compile": null, + "loader": { + "task": "depth-estimation", + "model_class": "AutoModelForDepthEstimation", + "model_type": "depth_pro" + } +} \ No newline at end of file