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()
|