unsupported cli model types

huggingface · Oct 21, 2024 · 08190c7 · 08190c7
1 parent 07d9952
commit 08190c7
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 5 deletions.
diff --git 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
@@ -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
@@ -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",