Spaces:
Running
Running
Commit History
-
Fixed ablation experiments, drilled down on which pre-loaded ones to run
3d71d04
-
Bug fix: type issue with models leading to gibberish outputs
5528a77
-
Rename L#-H# notation to L#-D# for detector consistency
Updates the head identifier labels from H (head) to D (detector) in ablation chips, results display, and pipeline detector labels to match the jargon-reduction renaming. Co-Authored-By: Claude Opus 4.6 <noreply@anthropic.com>deb2071
-
Added numbers to BertViz visualization, easier to map attention head numbers
67f5be3
-
added color swatches for easy mapping from attention categories to BertViz
96e2bc7 -
Fixed model-switching bug with activation store clearing
7ad3e7b
-
New models chosen and rag docs updated
f478beb
-
Attention refactor, better categorization and explanation
7fa8fb4
-
Chatbot improvements: UI cleaning, chat streaming, typing animation, and better markdown formatting
a684a8d -
ablation experiment fixes, better UI and full sequence comparison
a938eeb
-
feat(output): Add token scrubber with per-position top-5 next-token probabilities
Co-authored-by: Cursor <cursoragent@cursor.com>d60cfe2
-
Switch to cost-optimized paid models: Gemini 2.5 Flash Lite + text-embedding-3-small
Co-authored-by: Cursor <cursoragent@cursor.com>129c11e -
Switch to free models: Qwen3-Next chat + local embeddings
Co-authored-by: Cursor <cursoragent@cursor.com>94b077a -
Migrate from Gemini to OpenRouter API
Co-authored-by: Cursor <cursoragent@cursor.com>ef3e36a -
Migrate from google-generativeai to google-genai SDK
Co-authored-by: Cursor <cursoragent@cursor.com>ddd91a5 -
Resolve merge conflicts in README.md and todo.md
Co-authored-by: Cursor <cursoragent@cursor.com>b4d0227 -
feat: Add AI chatbot with Gemini integration and RAG support
Co-authored-by: Cursor <cursoragent@cursor.com>62884e7 -
cleanup for merge
b194610 -
chore: remove unused code, imports, and deprecated functions
- Delete components/tokenization_panel.py (superseded by pipeline.py) - Remove 6 unused imports from app.py - Remove deprecated _get_top_attended_tokens() and references - Remove unused create_stage_summary() from pipeline.py - Remove 7 unused utility functions from model_patterns.py and beam_search.py - Update utils/__init__.py exports and README.md Total: ~1,087 lines removed. All 81 tests pass. Co-authored-by: Cursor <cursoragent@cursor.com>2965a7d -
conductor setup
aabd66e
-
conductor(checkpoint): Checkpoint end of Phase 3: Visualization & Feedback Loop
72f82a4 -
conductor(checkpoint): Checkpoint end of Phase 2: Frontend Control Panel
24bd049 -
fix(utils): Export generate_bertviz_model_view_html from package
5cad664 -
conductor(checkpoint): Checkpoint end of Phase 1: Backend Support for Ablation
5fc7374 -
feat(pipeline): Add ablation support to beam search and verify pipeline dispatch
d2ea949 -
feat(ablation): Unify ablation execution into execute_forward_pass
2f4e4f3 -
Fix multi-layer ablation to ablate all heads simultaneously
ae08976
-
rules change for simple git commit messages
d90df7f -
Switch attention viz to head_view and add categorization (Agent D)
ef749ae
-
Add comprehensive test suite for utility modules (73 tests)
e13b1ba
-
Refactor: Replace heatmap with pipeline explanation visualization
3850656 -
heatmap implementation
ac745db
-
Ablation updated for full sequence, needs refactor for front-end and workflow
2ad1c2e
-
Fixed errors for beam search refactor
c19d5a8
-
Switched to beam search instead of single token
4c3d673 -
Fix storage quota error by using memory storage and add type safety checks
cc96787
-
Better explanations, moved transformer layer diagram to be more visible
080932d
-
Fix meta tensor error in safe_to_serializable
- Wrap device.type check in try-except to handle RuntimeError - Prevents 'Tensor on device meta is not on the expected device cpu!' error - Returns None for problematic meta tensors instead of crashing1f5efac -
Update layer highlighting to 100% threshold for actual output token only
- Changed detect_significant_probability_increases to only check actual output token instead of all top 5 tokens - Updated threshold from 50% to 100% (doubled probability) for highlighting layers - Modified compute_layer_wise_summaries to accept activation_data and extract actual output token - Updated all calls to compute_layer_wise_summaries in app.py to pass activation_data - Added _create_actual_output_display helper function to show actual model output - Display actual output token below final output graphs (normal mode, ablation mode, comparison mode) - Added tooltip explaining why actual output may differ from highest probability token in final layer due to residual connections - Updated graph tooltip text to reflect 100% threshold change9d22c3c
-
Improve explanations: optimize head categorization and comparison thresholds, enhance ablation explanation, rewrite all tooltips for high school level
b828e58 -
Clean up: fix significant layer threshold to 50%, remove unused logit lens parameter, remove certainty calculations
a4e6de8
-
Add ablation experiment before/after comparison with Reset button
- Added session-activation-store-original to preserve pre-ablation data - Modified run_head_ablation callback to save original data before ablation - Updated create_layer_accordions to detect ablation mode and show before/after sections for all layers - Modified _create_single_prompt_chart and _create_token_probability_delta_chart to accept title suffixes - Added before/after comparison for line graphs showing top 5 token probabilities across layers - Added Reset Ablation button in sidebar (visible only during ablation) - Implemented reset_ablation callback to restore original data and clear ablation - Added visual polish with section headers, background colors (blue for before, orange for after), and explanatory tooltips - All changes maintain backward compatibility with non-ablation moded701d46
-
Fix ablation: add attention_modules to session store so ablation can find attention layers
efbe4ad
-
Add comprehensive debugging to ablation data flow
a1ae11e -
Fix ablation: capture ablated attention module output to populate layer data
455f0c2 -
Refine layer organization: add title to diagram, move graph below layers, give Residual a box, reorder Explore button
34fa23c -
Reorganize Layer-by-Layer Predictions: move diagram to main container, reorder sections, update arrows, increase threshold to 75%
c127d17 -
Fix norm_selection type mismatch and ensure norm_parameters are always selected - Changed auto-selection to return list instead of string - Updated run_analysis to handle norm_patterns as list - Added norm_parameters requirement to Run Analysis button
b98d237 -
Clean up temporary documentation
f64fe64 -
Fix token predictions not being collected - changed condition to check for block_outputs and norm_parameters instead of logit_lens_parameter
d6f21db