Update download_model.py

Changed model to 7B
This commit is contained in:
James Devine 2023-08-29 13:09:10 +02:00 committed by GitHub
parent 2e1ee1e6f2
commit 9bb139be8b
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23

View file

@ -13,9 +13,9 @@ def download_mpt_quant(destination_folder: str, repo_id: str, model_filename: st
if __name__ == "__main__":
"""full url: https://huggingface.co/TheBloke/mpt-30B-chat-GGML/blob/main/mpt-30b-chat.ggmlv0.q4_1.bin"""
"""full url: https://huggingface.co/TheBloke/mpt-7b-chat-GGML/blob/main/mpt-7b-chat.ggmlv0.q4_1.bin"""
repo_id = "TheBloke/mpt-30B-chat-GGML"
model_filename = "mpt-30b-chat.ggmlv0.q4_1.bin"
repo_id = "TheBloke/mpt-7b-chat-GGML"
model_filename = "mpt-7b-chat.ggmlv0.q4_1.bin"
destination_folder = "models"
download_mpt_quant(destination_folder, repo_id, model_filename)