From 08190c70fbba5d89954a6ca81f49e30902300a65 Mon Sep 17 00:00:00 2001 From: IlyasMoutawwakil Date: Mon, 21 Oct 2024 17:09:30 +0200 Subject: [PATCH] unsupported cli model types --- optimum/exporters/tasks.py | 8 +++++--- tests/exporters/exporters_utils.py | 3 ++- tests/onnxruntime/utils_onnxruntime_tests.py | 2 +- 3 files changed, 8 insertions(+), 5 deletions(-) diff --git a/optimum/exporters/tasks.py b/optimum/exporters/tasks.py index 87ab62b2f2..82417886f2 100644 --- a/optimum/exporters/tasks.py +++ b/optimum/exporters/tasks.py @@ -1178,12 +1178,14 @@ class TasksManager: "transformers": _SUPPORTED_MODEL_TYPE, } _UNSUPPORTED_CLI_MODEL_TYPE = { - "unet", - "vae-encoder", - "vae-decoder", "clip-text-model", "clip-text-with-projection", + "sd3-transformer", + "t5-encoder", "trocr", # supported through the vision-encoder-decoder model type + "unet", + "vae-encoder", + "vae-decoder", } _SUPPORTED_CLI_MODEL_TYPE = ( set(_SUPPORTED_MODEL_TYPE.keys()) diff --git a/tests/exporters/exporters_utils.py b/tests/exporters/exporters_utils.py index c8a33b0be3..9617ab37a0 100644 --- a/tests/exporters/exporters_utils.py +++ b/tests/exporters/exporters_utils.py @@ -296,9 +296,10 @@ } PYTORCH_DIFFUSION_MODEL = { + "latent-consistency": "echarlaix/tiny-random-latent-consistency", "stable-diffusion": "hf-internal-testing/tiny-stable-diffusion-torch", + "stable-diffusion-3": "yujiepan/stable-diffusion-3-tiny-random", "stable-diffusion-xl": "echarlaix/tiny-random-stable-diffusion-xl", - "latent-consistency": "echarlaix/tiny-random-latent-consistency", } PYTORCH_TIMM_MODEL = { diff --git a/tests/onnxruntime/utils_onnxruntime_tests.py b/tests/onnxruntime/utils_onnxruntime_tests.py index 50eec6c95e..cb224993ad 100644 --- a/tests/onnxruntime/utils_onnxruntime_tests.py +++ b/tests/onnxruntime/utils_onnxruntime_tests.py @@ -107,10 +107,10 @@ "groupvit": "hf-internal-testing/tiny-random-groupvit", "hubert": "hf-internal-testing/tiny-random-HubertModel", "ibert": "hf-internal-testing/tiny-random-IBertModel", - "levit": "hf-internal-testing/tiny-random-LevitModel", "latent-consistency": "echarlaix/tiny-random-latent-consistency", "layoutlm": "hf-internal-testing/tiny-random-LayoutLMModel", "layoutlmv3": "hf-internal-testing/tiny-random-LayoutLMv3Model", + "levit": "hf-internal-testing/tiny-random-LevitModel", "longt5": "hf-internal-testing/tiny-random-LongT5Model", "llama": "optimum-internal-testing/tiny-random-llama", "m2m_100": "hf-internal-testing/tiny-random-m2m_100",