fix deepspeed

huggingface · Mar 27, 2024 · 675e30e · 675e30e
1 parent 855bf6f
commit 675e30e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/configs/_deepspeed_inference_.yaml b/tests/configs/_deepspeed_inference_.yaml
@@ -7,7 +7,7 @@ launcher:
 backend:
   device: cuda
   device_ids: 0,1
-  model: openai-community/gpt2
+  model: google-bert/bert-base-uncased
   deepspeed_inference: true
   deepspeed_inference_config:
     tensor_parallel: