Unverified Commit a36de682 authored by Philipp Moritz's avatar Philipp Moritz Committed by GitHub
Browse files

[Minor] Fix small typo in llama.py: QKVParallelLinear -> QuantizationConfig (#4991)

parent eb6d3c26
...@@ -57,7 +57,7 @@ class LlamaMLP(nn.Module): ...@@ -57,7 +57,7 @@ class LlamaMLP(nn.Module):
hidden_size: int, hidden_size: int,
intermediate_size: int, intermediate_size: int,
hidden_act: str, hidden_act: str,
quant_config: Optional[QKVParallelLinear] = None, quant_config: Optional[QuantizationConfig] = None,
bias: bool = False, bias: bool = False,
) -> None: ) -> None:
super().__init__() super().__init__()
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment