Spaces:
Runtime error
Runtime error
Commit
·
db24877
1
Parent(s):
19342c6
fix: alignment of parameters to generation function with advanced settings
Browse files- src/app.py +2 -2
src/app.py
CHANGED
|
@@ -48,9 +48,9 @@ def generate(
|
|
| 48 |
message: str,
|
| 49 |
chat_history: list[dict],
|
| 50 |
temperature: float = TEMPERATURE,
|
|
|
|
| 51 |
top_p: float = TOP_P,
|
| 52 |
top_k: float = TOP_K,
|
| 53 |
-
repetition_penalty: float = REPETITION_PENALTY,
|
| 54 |
max_new_tokens: int = MAX_NEW_TOKENS,
|
| 55 |
) -> Iterator[str]:
|
| 56 |
"""Generate function for chat demo."""
|
|
@@ -109,7 +109,7 @@ repetition_penalty_slider = gr.Slider(
|
|
| 109 |
minimum=0,
|
| 110 |
maximum=2.0,
|
| 111 |
value=REPETITION_PENALTY,
|
| 112 |
-
step=0.
|
| 113 |
label="Repetition Penalty",
|
| 114 |
elem_classes=["gr_accordion_element"],
|
| 115 |
)
|
|
|
|
| 48 |
message: str,
|
| 49 |
chat_history: list[dict],
|
| 50 |
temperature: float = TEMPERATURE,
|
| 51 |
+
repetition_penalty: float = REPETITION_PENALTY,
|
| 52 |
top_p: float = TOP_P,
|
| 53 |
top_k: float = TOP_K,
|
|
|
|
| 54 |
max_new_tokens: int = MAX_NEW_TOKENS,
|
| 55 |
) -> Iterator[str]:
|
| 56 |
"""Generate function for chat demo."""
|
|
|
|
| 109 |
minimum=0,
|
| 110 |
maximum=2.0,
|
| 111 |
value=REPETITION_PENALTY,
|
| 112 |
+
step=0.05,
|
| 113 |
label="Repetition Penalty",
|
| 114 |
elem_classes=["gr_accordion_element"],
|
| 115 |
)
|