Monster commited on
Commit
d2f4738
1 Parent(s): 9e11258

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
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", n_ctx=2048, seed=0)
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
- for x in llm(ins.format(instruction), stop=['### Instruction:', '### End'], stream=True):
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