model change
Browse files
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" #
|
9 |
-
|
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(
|