This commit is contained in:
Andrei Betlen 2023-08-08 15:07:28 -04:00
parent 843b7ccd90
commit 1e844d3238

View file

@ -230,7 +230,7 @@ class Llama:
grammar: Optional[Union[str, Path]] = None,
n_gqa: Optional[int] = None, # (TEMPORARY) must be 8 for llama2 70b
rms_norm_eps: Optional[float] = None, # (TEMPORARY)
mul_mat_q: Optional(bool) = None, # (TEMPORARY)
mul_mat_q: Optional[bool] = None, # (TEMPORARY)
verbose: bool = True,
):
"""Load a llama.cpp model from `model_path`.