Update config/config.py

config/config.py

@@ -1,4 +1,40 @@
-# config/config.py
+# config/config.py 
+
+from dataclasses import dataclass, field
+from enum import Enum
+from typing import Dict, Any, Optional
+
+class GenerationStrategy(str, Enum):
+    DEFAULT = "default"
+    MAJORITY_VOTING = "majority_voting"
+    BEST_OF_N = "best_of_n"
+    BEAM_SEARCH = "beam_search"
+    DVTS = "dvts"
+    COT = "chain_of_thought"
+    REACT = "react"
+
+@dataclass
+class ModelConfig:
+    model_kwargs: Dict[str, Any] = field(default_factory=dict)
+    tokenizer_kwargs: Dict[str, Any] = field(default_factory=dict)
+    quantization_kwargs: Dict[str, Any] = field(default_factory=dict)
+
+@dataclass
+class GenerationConfig:
+    num_samples: int = 5
+    depth: int = 3
+    breadth: int = 2
+    max_history_turns: int = 1
+    max_new_tokens: int = 50
+    temperature: float = 0.7
+    top_p: float = 0.9
+    top_k: int = 50
+    repetition_penalty: float = 1.1
+    length_penalty: float = 1.0
+    do_sample: bool = True
+    strategy: GenerationStrategy = GenerationStrategy.DEFAULT
+
+#####
 from pydantic_settings import BaseSettings
 from pathlib import Path
 import torch