update usage
Browse files
README.md
CHANGED
|
@@ -42,12 +42,11 @@ import PIL
|
|
| 42 |
from clipscope import ConfiguredViT, TopKSAE
|
| 43 |
|
| 44 |
device='cpu'
|
| 45 |
-
filename_in_hf_repo = "
|
| 46 |
-
sae = TopKSAE.from_pretrained(
|
| 47 |
|
| 48 |
-
transformer_name='laion/CLIP-ViT-L-14-laion2B-s32B-b82K'
|
| 49 |
locations = [(22, 'resid')]
|
| 50 |
-
transformer = ConfiguredViT(locations,
|
| 51 |
|
| 52 |
input = PIL.Image.new("RGB", (224, 224), (0, 0, 0)) # black image for testing
|
| 53 |
|
|
|
|
| 42 |
from clipscope import ConfiguredViT, TopKSAE
|
| 43 |
|
| 44 |
device='cpu'
|
| 45 |
+
filename_in_hf_repo = "22_resid/1200013184.pt"
|
| 46 |
+
sae = TopKSAE.from_pretrained(checkpoint=filename_in_hf_repo, device=device)
|
| 47 |
|
|
|
|
| 48 |
locations = [(22, 'resid')]
|
| 49 |
+
transformer = ConfiguredViT(locations, device=device)
|
| 50 |
|
| 51 |
input = PIL.Image.new("RGB", (224, 224), (0, 0, 0)) # black image for testing
|
| 52 |
|