Di Zhang
commited on
Update app.py
Browse files
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,
|
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}")
|