File size: 321 Bytes
4b31e86
 
 
 
 
 
51dd3b2
 
 
4b31e86
51dd3b2
 
1
2
3
4
5
6
7
8
9
10
11
12
13
# ecoeval/config.py
from dataclasses import dataclass


@dataclass
class EcoEvalConfig:
    model_id: str = "Salesforce/codegen-350M-mono"
    max_new_tokens: int = 96      # a bit shorter to reduce rambling
    temperature: float = 0.0      # deterministic, less chatty
    top_p: float = 0.95
    device: str = "auto"