Update app.py

app.py

@@ -2519,7 +2519,9 @@ def run_ablation_study(model_name: str, base_config: CompressionConfig, dataset_
         test_configs.append(("stage1_only", {
             "compression_type": CompressionType.ENHANCED_SPG,
             "enable_two_stage": True,
-            "stage2_compression_ratio": 1.0,  # Effectively disable Stage 2
+            "stage2_compression_ratio": 1.01,  # Minimal Stage 2 (effectively disabled)
+            "use_hybrid_sparse_attention": False,  # Disable HSA for Stage 2
+            "enable_head_compression": False,  # Disable head compression
             "description": "Stage 1 only"
         }))
     
@@ -2527,7 +2529,8 @@ def run_ablation_study(model_name: str, base_config: CompressionConfig, dataset_
         test_configs.append(("stage2_only", {
             "compression_type": CompressionType.ENHANCED_SPG,
             "enable_two_stage": True,
-            "stage1_compression_ratio": 1.0,  # Effectively disable Stage 1
+            "stage1_compression_ratio": 1.01,  # Minimal Stage 1 (effectively disabled)
+            "use_snapkv_plus_plus": False,  # Disable SnapKV++ for Stage 1
             "description": "Stage 2 only"
         }))