Spaces:
Running
on
Zero
Running
on
Zero
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 |
)
|