noanabeshima
commited on
Commit
•
8a08580
1
Parent(s):
32b74e7
Update README.md
Browse files
README.md
CHANGED
@@ -14,7 +14,7 @@ def load_feat_acts(fname, only_active_docs=False):
|
|
14 |
local_path = hf_hub_download(repo_id="noanabeshima/tiny_model_cached_acts", filename=fname)
|
15 |
csr_kwargs = torch.load(local_path)
|
16 |
|
17 |
-
# The matrices
|
18 |
# Convert them back before constructing the matrix.
|
19 |
csr_kwargs['crow_indices'] = csr_kwargs['crow_indices'].int()
|
20 |
csr_kwargs['col_indices'] = csr_kwargs['col_indices'].int()
|
|
|
14 |
local_path = hf_hub_download(repo_id="noanabeshima/tiny_model_cached_acts", filename=fname)
|
15 |
csr_kwargs = torch.load(local_path)
|
16 |
|
17 |
+
# The matrices might be stored in space-efficient formats that're incompatible with torch's sparse csr tensor.
|
18 |
# Convert them back before constructing the matrix.
|
19 |
csr_kwargs['crow_indices'] = csr_kwargs['crow_indices'].int()
|
20 |
csr_kwargs['col_indices'] = csr_kwargs['col_indices'].int()
|