Spaces:
Running
Running
Update ecoeval/config.py
Browse files- ecoeval/config.py +5 -4
ecoeval/config.py
CHANGED
|
@@ -4,8 +4,9 @@ from dataclasses import dataclass
|
|
| 4 |
|
| 5 |
@dataclass
|
| 6 |
class EcoEvalConfig:
|
| 7 |
-
model_id: str = "Salesforce/codegen-350M-
|
| 8 |
-
max_new_tokens: int =
|
| 9 |
-
temperature: float = 0.
|
| 10 |
top_p: float = 0.95
|
| 11 |
-
device: str = "auto"
|
|
|
|
|
|
| 4 |
|
| 5 |
@dataclass
|
| 6 |
class EcoEvalConfig:
|
| 7 |
+
model_id: str = "Salesforce/codegen-350M-mono"
|
| 8 |
+
max_new_tokens: int = 96 # a bit shorter to reduce rambling
|
| 9 |
+
temperature: float = 0.0 # deterministic, less chatty
|
| 10 |
top_p: float = 0.95
|
| 11 |
+
device: str = "auto"
|
| 12 |
+
|