Di Zhang commited on
Commit
925e2eb
·
verified ·
1 Parent(s): 19b2d9c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -37,7 +37,7 @@ def generate_text(message, history, max_tokens=512, temperature=0.9, top_p=0.95)
37
  input_texts = [input_text.replace('<|end_of_text|>','') for input_text in input_texts]
38
  #print(f"input_texts[0]: {input_texts[0]}")
39
  inputs = model.tokenize(input_texts[0].encode('utf-8'))
40
- for token in model(inputs, top_p=top_p, temperature=temperature, stream=True):
41
  #print(f"token: {token}")
42
  text = model.detokenize([token])
43
  #print(f"text detok: {text}")
 
37
  input_texts = [input_text.replace('<|end_of_text|>','') for input_text in input_texts]
38
  #print(f"input_texts[0]: {input_texts[0]}")
39
  inputs = model.tokenize(input_texts[0].encode('utf-8'))
40
+ for token in model.generate(inputs, top_p=top_p, temp=temperature):
41
  #print(f"token: {token}")
42
  text = model.detokenize([token])
43
  #print(f"text detok: {text}")