change app to low model 15m
Browse files
app.py
CHANGED
@@ -35,8 +35,8 @@ class Prompter(object):
|
|
35 |
|
36 |
|
37 |
load_8bit = False # for Colab
|
38 |
-
base_model = "
|
39 |
-
lora_weights = "
|
40 |
share_gradio = True
|
41 |
|
42 |
if torch.cuda.is_available():
|
|
|
35 |
|
36 |
|
37 |
load_8bit = False # for Colab
|
38 |
+
base_model = "nickypro/tinyllama-15M"
|
39 |
+
lora_weights = "./chkp"
|
40 |
share_gradio = True
|
41 |
|
42 |
if torch.cuda.is_available():
|