Update app.py
Browse files
app.py
CHANGED
@@ -24,7 +24,7 @@ examples = [
|
|
24 |
]
|
25 |
|
26 |
model_name = "berkeley-nest/Starling-RM-7B-alpha"
|
27 |
-
base_model = "
|
28 |
|
29 |
|
30 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|
|
|
24 |
]
|
25 |
|
26 |
model_name = "berkeley-nest/Starling-RM-7B-alpha"
|
27 |
+
base_model = "meta-llama/Llama-2-7b-chat-hf"
|
28 |
|
29 |
|
30 |
device = "cuda" if torch.cuda.is_available() else "cpu"
|