fix
Browse files- translate.py +1 -1
translate.py
CHANGED
|
@@ -38,7 +38,7 @@ stop_criteria_list = transformers.StoppingCriteriaList(
|
|
| 38 |
)
|
| 39 |
|
| 40 |
|
| 41 |
-
def chat(
|
| 42 |
prompt = f"<|im_start|>user\n{q}<|im_end|>\n<|im_start|>assistant"
|
| 43 |
old_tids = tokenizer.encode(prompt)
|
| 44 |
|
|
|
|
| 38 |
)
|
| 39 |
|
| 40 |
|
| 41 |
+
def chat(q, temperature = 0.5):
|
| 42 |
prompt = f"<|im_start|>user\n{q}<|im_end|>\n<|im_start|>assistant"
|
| 43 |
old_tids = tokenizer.encode(prompt)
|
| 44 |
|