singhalamaan116 commited on
Commit
51dd3b2
·
verified ·
1 Parent(s): cb2d7b5

Update ecoeval/config.py

Browse files
Files changed (1) hide show
  1. 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-multi"
8
- max_new_tokens: int = 128
9
- temperature: float = 0.2
10
  top_p: float = 0.95
11
- device: str = "auto" # "cpu", "cuda", or "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
+