added timm tests (tiny vit)

huggingface · Jan 13, 2024 · 4022aca · 4022aca
1 parent e8b4e2b
commit 4022aca
Show file tree

Hide file tree

Showing 7 changed files with 34 additions and 21 deletions.
diff --git a/tests/configs/_diffusers_.yaml b/tests/configs/_diffusers_.yaml
@@ -0,0 +1,6 @@
+model: hf-internal-testing/tiny-stable-diffusion-torch
+task: stable-diffusion
+
+benchmark:
+  forward_kwargs:
+    num_inference_steps: 2
diff --git a/tests/configs/_timm_.yaml b/tests/configs/_timm_.yaml
@@ -0,0 +1,2 @@
+model: timm/tiny_vit_21m_224.in1k
+task: image_classification
diff --git a/.../cpu_inference_onnxruntime_diffusion.yaml → .../cpu_inference_onnxruntime_diffusers.yaml b/.../cpu_inference_onnxruntime_diffusion.yaml → .../cpu_inference_onnxruntime_diffusers.yaml
@@ -3,13 +3,8 @@ defaults:
   # order of inheritance, last one overrides previous ones
   - _base_ # inherits from base config
   - _inference_ # inherits from inference config
+  - _diffusers_ # inherits from diffusers config
   - _self_ # for hydra 1.1 compatibility
 
-experiment_name: cpu_inference_onnxruntime_diffusion
-model: hf-internal-testing/tiny-stable-diffusion-torch
-task: stable-diffusion
+experiment_name: cpu_inference_onnxruntime_diffusers
 device: cpu
-
-benchmark:
-  forward_kwargs:
-    num_inference_steps: 2
diff --git a/tests/configs/cpu_inference_onnxruntime_timm.yaml b/tests/configs/cpu_inference_onnxruntime_timm.yaml
@@ -0,0 +1,10 @@
+defaults:
+  - backend: onnxruntime
+  # order of inheritance, last one overrides previous ones
+  - _base_ # inherits from base config
+  - _inference_ # inherits from inference config
+  - _timm_ # inherits from timm config
+  - _self_ # for hydra 1.1 compatibility
+
+experiment_name: cpu_inference_onnxruntime_timm
+device: cpu
diff --git a/...igs/cpu_inference_openvino_diffusion.yaml → ...igs/cpu_inference_openvino_diffusers.yaml b/...igs/cpu_inference_openvino_diffusion.yaml → ...igs/cpu_inference_openvino_diffusers.yaml
@@ -3,13 +3,8 @@ defaults:
   # order of inheritance, last one overrides previous ones
   - _base_ # inherits from base config
   - _inference_ # inherits from inference config
+  - _diffusers_ # inherits from diffusers config
   - _self_ # for hydra 1.1 compatibility
 
-experiment_name: cpu_inference_openvino_diffusion
-model: hf-internal-testing/tiny-stable-diffusion-torch
-task: stable-diffusion
+experiment_name: cpu_inference_openvino_diffusers
 device: cpu
-
-benchmark:
-  forward_kwargs:
-    num_inference_steps: 2
diff --git a/...figs/cpu_inference_pytorch_diffusion.yaml → ...figs/cpu_inference_pytorch_diffusers.yaml b/...figs/cpu_inference_pytorch_diffusion.yaml → ...figs/cpu_inference_pytorch_diffusers.yaml
@@ -3,13 +3,8 @@ defaults:
   # order of inheritance, last one overrides previous ones
   - _base_ # inherits from base config
   - _inference_ # inherits from inference config
+  - _diffusers_ # inherits from diffusers config
   - _self_ # for hydra 1.1 compatibility
 
-experiment_name: cpu_inference_pytorch_diffusion
-model: hf-internal-testing/tiny-stable-diffusion-torch
-task: stable-diffusion
+experiment_name: cpu_inference_pytorch_diffusers
 device: cpu
-
-benchmark:
-  forward_kwargs:
-    num_inference_steps: 2
diff --git a/tests/configs/cpu_inference_pytorch_timm.yaml b/tests/configs/cpu_inference_pytorch_timm.yaml
@@ -0,0 +1,10 @@
+defaults:
+  - backend: pytorch
+  # order of inheritance, last one overrides previous ones
+  - _base_ # inherits from base config
+  - _inference_ # inherits from inference config
+  - _timm_ # inherits from timm config
+  - _self_ # for hydra 1.1 compatibility
+
+experiment_name: cpu_inference_pytorch_timm
+device: cpu
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1,2 @@
		model: timm/tiny_vit_21m_224.in1k
		task: image_classification