change the worker to 3 (deepjavalibrary#533)

Lokiiiiii · Mar 8, 2023 · 9ae1e0c · 9ae1e0c
1 parent 525dc58
commit 9ae1e0c
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tests/integration/llm/client.py b/tests/integration/llm/client.py
@@ -130,7 +130,7 @@
 
 transformers_neuronx_model_spec = {
  "opt-1.3b": {
- "worker": 2,
+ "worker": 3,
  "seq_length": [128, 256],
  "batch_size": [4]
  }