Add llama-2-70b GGML support (#3285)

This commit is contained in:
oobabooga 2023-07-24 16:37:03 -03:00 committed by GitHub
parent 6f4830b4d3
commit a07d070b6c
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
8 changed files with 20 additions and 4 deletions

View file

@ -30,6 +30,8 @@ loaders_and_params = {
],
'llama.cpp': [
'n_ctx',
'n_gqa',
'rms_norm_eps',
'n_gpu_layers',
'n_batch',
'threads',
@ -42,6 +44,8 @@ loaders_and_params = {
],
'llamacpp_HF': [
'n_ctx',
'n_gqa',
'rms_norm_eps',
'n_gpu_layers',
'n_batch',
'threads',