Spaces:
Runtime error
Runtime error
project-baize
commited on
Commit
•
fdf1396
1
Parent(s):
a56a74d
Update app.py
Browse files
app.py
CHANGED
@@ -15,7 +15,7 @@ logging.basicConfig(
|
|
15 |
|
16 |
base_model = "decapoda-research/llama-7b-hf"
|
17 |
adapter_model = "project-baize/baize-lora-7B"
|
18 |
-
|
19 |
|
20 |
|
21 |
def predict(text,
|
|
|
15 |
|
16 |
base_model = "decapoda-research/llama-7b-hf"
|
17 |
adapter_model = "project-baize/baize-lora-7B"
|
18 |
+
tokenizer,model,device = load_tokenizer_and_model(base_model,adapter_model)
|
19 |
|
20 |
|
21 |
def predict(text,
|