We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent 690f471 commit 52e62c0Copy full SHA for 52e62c0
llm/predict/predictor.py
@@ -104,7 +104,7 @@ class PredictorArgument:
104
default="fp16",
105
metadata={"help": "avx cachekv type. Supported values: fp16,int8"},
106
)
107
- batch_size: int = field(default=10, metadata={"help": "The batch size of data."})
+ batch_size: int = field(default=1, metadata={"help": "The batch size of data."})
108
benchmark: bool = field(
109
default=False,
110
metadata={
0 commit comments