fix

huggingface · Nov 22, 2023 · 60be0c2 · 60be0c2
1 parent 4629948
commit 60be0c2
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/tests/configs/_tp_.yaml b/tests/configs/_tp_.yaml
@@ -1,6 +1,6 @@
 # Tensor Parallelism (TP)
 defaults:
-  - launcher: torchrun # for TP training, use torchrun launcher
+  - override launcher: torchrun # for TP training, use torchrun launcher
 
 hydra:
   job:

diff --git a/tests/configs/cuda_pytorch_inference_bert_tp.yaml b/tests/configs/cuda_pytorch_inference_bert_tp.yaml
@@ -5,6 +5,6 @@ defaults:
   - _tp_ # inherits from tensor parallelism config
   - _self_ # for hydra 1.1 compatibility
 
-model: hf-internal-testing/tiny-random-bert
+model: bert-base-uncased
 task: text-classification
 device: cuda