ai-agent-book 精选快照（<2MB 代码与文档，来自 github.com/bojieli/ai-agent-book）

chapter2/attention_visualization/env.example

@@ -0,0 +1,34 @@
+# Model Configuration
+MODEL_NAME=Qwen/Qwen3-0.6B
+# MODEL_PATH=/path/to/local/model  # Optional: Use local model
+
+# Device Configuration
+DEVICE=auto  # auto, cuda, mps, or cpu
+
+# Attention Configuration
+ATTENTION_LAYER_INDEX=-1  # -1 for last layer, or specify layer index
+TRACK_ALL_LAYERS=false    # Set to true to track all layers (memory intensive)
+
+# Generation Configuration
+MAX_NEW_TOKENS=100
+TEMPERATURE=0.7
+TOP_P=0.9
+REPETITION_PENALTY=1.1
+
+# Visualization Configuration
+VIZ_OUTPUT_DIR=visualizations
+VIZ_FORMATS=heatmap,flow,summary
+VIZ_COLORMAP=viridis
+VIZ_FIGSIZE=14,10
+VIZ_DPI=150
+
+# Logging
+LOG_LEVEL=INFO
+LOG_FILE=attention_viz.log
+
+# Output
+RESULTS_DIR=results
+
+# Interactive Mode
+INTERACTIVE_MODE=true
+AUTO_VISUALIZE=true