fix py-txi

huggingface · Nov 27, 2024 · 5ee4982 · 5ee4982
1 parent 86634fa
commit 5ee4982
Show file tree

Hide file tree

Showing 4 changed files with 8 additions and 5 deletions.
diff --git a/.github/workflows/test_cli_cuda_pytorch.yaml b/.github/workflows/test_cli_cuda_pytorch.yaml
@@ -74,7 +74,7 @@ jobs:
 
       - name: Install dependencies
         run: |
-          pip install -e .[testing,diffusers,timm,peft,deepspeed]
+          pip install -e .[testing,diffusers,timm,peft,deepspeed] -U
 
       - name: Run tests (parallel)
         run: |

diff --git a/tests/configs/_st_bert_.yaml b/tests/configs/_st_bert_.yaml
@@ -0,0 +1,3 @@
+backend:
+  model: sentence-transformers/all-MiniLM-L6-v2
+  task: feature-extraction
diff --git a/tests/configs/cpu_inference_py_txi_bert.yaml → ...configs/cpu_inference_py_txi_st_bert.yaml b/tests/configs/cpu_inference_py_txi_bert.yaml → ...configs/cpu_inference_py_txi_st_bert.yaml
@@ -3,8 +3,8 @@ defaults:
   - _base_ # inherits from base config
   - _cpu_ # inherits from cpu config
   - _inference_ # inherits from inference config
-  - _bert_ # inherits from bert config
+  - _st_bert_ # inherits from bert config
   - _self_ # hydra 1.1 compatibility
   - override backend: py-txi
 
-name: cpu_inference_py_txi_bert
+name: cpu_inference_py_txi_st_bert
diff --git a/...s/configs/cuda_inference_py_txi_bert.yaml → ...onfigs/cuda_inference_py_txi_st_bert.yaml b/...s/configs/cuda_inference_py_txi_bert.yaml → ...onfigs/cuda_inference_py_txi_st_bert.yaml
@@ -3,8 +3,8 @@ defaults:
   - _base_ # inherits from base config
   - _cuda_ # inherits from cuda config
   - _inference_ # inherits from inference config
-  - _bert_ # inherits from bert config
+  - _st_bert_ # inherits from bert config
   - _self_ # hydra 1.1 compatibility
   - override backend: py-txi
 
-name: cuda_inference_py_txi_bert
+name: cuda_inference_py_txi_st_bert