Anson69 commited on
Commit
aeb430f
1 Parent(s): 580c70e

model change

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,8 +5,8 @@ import json
5
  import os
6
 
7
  # model_name = "meta-llama/Meta-Llama-3-8B-Instruct" # uses more than 16GB
8
- model_name = "meta-llama/Llama-2-7b-chat-hf" # try this with quantisation to reduce memory usage
9
- # model_name = "mistralai/Mistral-7B-Instruct-v0.1" # try this with quantisation to reduce memory usage
10
 
11
 
12
  quantization_config = BitsAndBytesConfig(
 
5
  import os
6
 
7
  # model_name = "meta-llama/Meta-Llama-3-8B-Instruct" # uses more than 16GB
8
+ # model_name = "meta-llama/Llama-2-7b-chat-hf" # only part of paid model
9
+ model_name = "mistralai/Mistral-7B-Instruct-v0.1" # try this with quantisation to reduce memory usage
10
 
11
 
12
  quantization_config = BitsAndBytesConfig(