Skip to content

Commit

Permalink
remove hanging streaming test case (deepjavalibrary#618)
Browse files Browse the repository at this point in the history
  • Loading branch information
rohithkrn authored Apr 12, 2023
1 parent f8fd03a commit c93f22a
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tests/integration/llm/client.py
Original file line number Diff line number Diff line change
Expand Up @@ -82,7 +82,7 @@
"gpt-neo-1.3b": {
"max_memory_per_gpu": [4.0, 5.0],
"batch_size": [1, 4],
"seq_length": [16, 32],
"seq_length": [16],
"worker": 2,
"stream_output": True,
}
Expand Down

0 comments on commit c93f22a

Please sign in to comment.