ffreemt commited on
Commit
9dbbc8e
1 Parent(s): 3f10600

Update top_k=50

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -106,7 +106,7 @@ ns = SimpleNamespace(
106
  @dataclass
107
  class GenerationConfig:
108
  temperature: float = 0.7
109
- top_k: int = 0
110
  top_p: float = 0.9
111
  repetition_penalty: float = 1.0
112
  max_new_tokens: int = 512
 
106
  @dataclass
107
  class GenerationConfig:
108
  temperature: float = 0.7
109
+ top_k: int = 50
110
  top_p: float = 0.9
111
  repetition_penalty: float = 1.0
112
  max_new_tokens: int = 512