Update config.py

config.py

@@ -334,9 +334,15 @@ class CompressionConfig:
     
     # Dataset configuration
     dataset_name: str = "wikitext"
-    dataset_config: str = "wikitext-2-raw-v1"
+    dataset_config: str = "wikitext-103-raw-v1"
     dataset_split: str = "test"
     
+    # Model configuration for publication
+    model_name: str = "gpt2"
+    test_sequence_lengths: List[int] = field(default_factory=lambda: [2048, 4096, 8192, 16384])
+    downstream_tasks: List[str] = field(default_factory=lambda: ["perplexity", "gsm8k", "mmlu"])
+    baseline_methods: List[str] = field(default_factory=lambda: ["h2o", "streamingllm", "snapkv"])
+    
     # Memory and system settings
     clear_cache_between_runs: bool = True
     use_memory_snapshot: bool = True