kazuma313 commited on
Commit
5b85bb3
1 Parent(s): be53556

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +13 -13
app.py CHANGED
@@ -31,20 +31,20 @@ def output_inference(tanya, history):
31
  prompt = prompt_template.format(ask=tanya)
32
 
33
  output = llm(
34
- repeat_penalty=1.2,
35
- prompt,
36
- stop=["<|end_of_text|>"],
37
- max_tokens=512,
38
- temperature=0.3,
39
- top_p=0.95,
40
- top_k=40,
41
- min_p=0.05,
42
- typical_p=1.0,
43
- stream=True)
44
  for out in output:
45
- stream = copy.deepcopy(out)
46
- temp += stream["choices"][0]["text"]
47
- yield temp
48
 
49
  history = ["init", prompt]
50
 
 
31
  prompt = prompt_template.format(ask=tanya)
32
 
33
  output = llm(
34
+ prompt,
35
+ stop=["<|end_of_text|>"],
36
+ max_tokens=512,
37
+ temperature=0.3,
38
+ top_p=0.95,
39
+ top_k=40,
40
+ min_p=0.05,
41
+ typical_p=1.0,
42
+ stream=True,
43
+ )
44
  for out in output:
45
+ stream = copy.deepcopy(out)
46
+ temp += stream["choices"][0]["text"]
47
+ yield temp
48
 
49
  history = ["init", prompt]
50