Fix the docstring
Browse files- eval_utils.py +1 -1
eval_utils.py
CHANGED
|
@@ -57,7 +57,7 @@ def predict_top_p(model, tokenizer, text, P=0.9):
|
|
| 57 |
|
| 58 |
|
| 59 |
def prompt_chat_LLM(model, tokenizer, text):
|
| 60 |
-
"""Prompt the
|
| 61 |
predicted_dialects = []
|
| 62 |
for dialect in DIALECTS_WITH_LABELS:
|
| 63 |
messages = [
|
|
|
|
| 57 |
|
| 58 |
|
| 59 |
def prompt_chat_LLM(model, tokenizer, text):
|
| 60 |
+
"""Prompt the model to determine whether the input text is acceptable in each of the 11 dialects."""
|
| 61 |
predicted_dialects = []
|
| 62 |
for dialect in DIALECTS_WITH_LABELS:
|
| 63 |
messages = [
|