Change app
Browse files
app.py
CHANGED
@@ -34,7 +34,7 @@ class Prompter(object):
|
|
34 |
)
|
35 |
|
36 |
|
37 |
-
load_8bit =
|
38 |
base_model = "baffo32/decapoda-research-llama-7B-hf"
|
39 |
lora_weights = "tiedong/goat-lora-7b"
|
40 |
share_gradio = True
|
|
|
34 |
)
|
35 |
|
36 |
|
37 |
+
load_8bit = False # for Colab
|
38 |
base_model = "baffo32/decapoda-research-llama-7B-hf"
|
39 |
lora_weights = "tiedong/goat-lora-7b"
|
40 |
share_gradio = True
|