Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -8,7 +8,7 @@ from llama_cpp import Llama
|
|
8 |
from huggingface_hub import hf_hub_download
|
9 |
|
10 |
hf_hub_download(repo_id="Pi3141/alpaca-lora-7B-ggml", filename="ggml-model-q4_1.bin", local_dir=".")
|
11 |
-
llm = Llama(model_path="./ggml-model-q4_1.bin",
|
12 |
|
13 |
|
14 |
ins = '''Below is an instruction that describes a task. Write a response that appropriately completes the request.
|
@@ -39,7 +39,7 @@ theme = gr.themes.Monochrome(
|
|
39 |
|
40 |
def generate(instruction):
|
41 |
result = ""
|
42 |
-
|
43 |
result += x['choices'][0]['text']
|
44 |
yield result
|
45 |
|
|
|
8 |
from huggingface_hub import hf_hub_download
|
9 |
|
10 |
hf_hub_download(repo_id="Pi3141/alpaca-lora-7B-ggml", filename="ggml-model-q4_1.bin", local_dir=".")
|
11 |
+
llm = Llama(model_path="./ggml-model-q4_1.bin", seed=0)
|
12 |
|
13 |
|
14 |
ins = '''Below is an instruction that describes a task. Write a response that appropriately completes the request.
|
|
|
39 |
|
40 |
def generate(instruction):
|
41 |
result = ""
|
42 |
+
for x in llm(ins.format(instruction), stop=['### Instruction:', '### End'], stream=True, max_tokens=1024):
|
43 |
result += x['choices'][0]['text']
|
44 |
yield result
|
45 |
|