grahamwhiteuk commited on
Commit
db24877
·
1 Parent(s): 19342c6

fix: alignment of parameters to generation function with advanced settings

Browse files
Files changed (1) hide show
  1. 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.1,
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
  )