Skip to content

Commit

Permalink
Experimenting with Llama-2
Browse files Browse the repository at this point in the history
  • Loading branch information
KillianLucas committed Aug 23, 2023
1 parent 6b15799 commit 1ed362c
Showing 1 changed file with 2 additions and 2 deletions.
4 changes: 2 additions & 2 deletions interpreter/llama_2.py
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,7 @@
def get_llama_2_instance():

# Define the file name
file_name = "llama-2-7b-chat.ggmlv3.q4_0.bin"
file_name = "llama-2-7b-chat.ggmlv3.q2_K.bin"

# Get user data directory for your application
user_data_dir = appdirs.user_data_dir("open-interpreter")
Expand Down Expand Up @@ -38,7 +38,7 @@ def get_llama_2_instance():
download_path = os.path.join(default_path, file_name)
message = f"Llama-2 not found. Would you like to download the `6.9GB` file to `{download_path}`?"
if confirm_action(message):
url = "https://huggingface.co/TheBloke/Llama-2-7B-chat-GGML/resolve/main/llama-2-7b-chat.ggmlv3.q4_0.bin"
url = "https://huggingface.co/TheBloke/Llama-2-7B-chat-GGML/resolve/main/llama-2-7b-chat.ggmlv3.q2_K.bin"
subprocess.run(f"curl -L '{url}' -o '{download_path}'", shell=True)
model_path = download_path
print('\n', "Finished downloading Llama-2.", '\n')
Expand Down

0 comments on commit 1ed362c

Please sign in to comment.