Spaces:

lmms-lab
/

Multimodal-SAE

Running on Zero

kcz358 commited on Mar 3

Commit

88f8ab3

1 Parent(s): 76d48cf

Remove gpu for vis

Files changed (1) hide show

app.py CHANGED Viewed

@@ -93,7 +93,6 @@ def generate_activations(image):
     torch.cuda.empty_cache()
     return topk_indices
-@spaces.GPU
 def visualize_activations(image, feature_num):
     global cached_tensor
     base_img_tokens = 576

     torch.cuda.empty_cache()
     return topk_indices
 def visualize_activations(image, feature_num):
     global cached_tensor
     base_img_tokens = 576