diff --git a/llama2.py b/llama2.py index 5618440..9cc4483 100644 --- a/llama2.py +++ b/llama2.py @@ -78,7 +78,7 @@ class LlamaCpp(PyOperator): 'filename': 'llama-2-7b.ggmlv3.q4_0.bin' }, 'llama-2-13b-chat': { - 'hf_id': 'TheBloke/Llama-2-13B-GGML', + 'hf_id': 'TheBloke/Llama-2-13B-chat-GGML', 'filename': 'llama-2-13b-chat.ggmlv3.q4_0.bin' } }