Spaces:

lmms-lab
/

Multimodal-SAE

Running on Zero

kcz358 commited on Mar 3

Commit

bfeec53

1 Parent(s): a8af4c9

Update

Files changed (1) hide show

app.py CHANGED Viewed

@@ -94,8 +94,9 @@ def generate_activations(image):
     torch.cuda.empty_cache()
     return topk_indices
 def visualize_activations(image, feature_num):
     base_img_tokens = 576
     patch_size = 24
     # Using Cached tensor

     torch.cuda.empty_cache()
     return topk_indices
+@spaces.GPU
 def visualize_activations(image, feature_num):
+    global cached_tensor
     base_img_tokens = 576
     patch_size = 24
     # Using Cached tensor