kcz358 commited on
Commit
7bcf3b0
·
1 Parent(s): 2a1ed01

hardcode unrelated features

Browse files
Files changed (1) hide show
  1. app.py +5 -1
app.py CHANGED
@@ -49,6 +49,10 @@ greedy_file_path = "assets/greedy.jpg"
49
  railway_file_path = "assets/railway.jpg"
50
  happy_file_path = "assets/happy.jpg"
51
 
 
 
 
 
52
  @spaces.GPU
53
  def generate_activations(image):
54
  prompt = "<image>"
@@ -96,7 +100,7 @@ def generate_activations(image):
96
  handle.remove()
97
  examples = []
98
  for indice in topk_indices:
99
- if indice <= 5000:
100
  examples.append([ f"model.layers.24_feature{indice.item()}",explanations[f"model.layers.24_feature{indice.item()}"]])
101
 
102
 
 
49
  railway_file_path = "assets/railway.jpg"
50
  happy_file_path = "assets/happy.jpg"
51
 
52
+ unrelated_features = [
53
+ 693, 793, 2033, 1195, 4879, 890, 1762, 45, 4258
54
+ ]
55
+
56
  @spaces.GPU
57
  def generate_activations(image):
58
  prompt = "<image>"
 
100
  handle.remove()
101
  examples = []
102
  for indice in topk_indices:
103
+ if indice <= 5000 and indice not in unrelated_features:
104
  examples.append([ f"model.layers.24_feature{indice.item()}",explanations[f"model.layers.24_feature{indice.item()}"]])
105
 
106