Spaces:
Runtime error
Runtime error
ffreemt
commited on
Commit
·
9dbbc8e
1
Parent(s):
3f10600
Update top_k=50
Browse files
app.py
CHANGED
@@ -106,7 +106,7 @@ ns = SimpleNamespace(
|
|
106 |
@dataclass
|
107 |
class GenerationConfig:
|
108 |
temperature: float = 0.7
|
109 |
-
top_k: int =
|
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
|