noanabeshima commited on
Commit
8a08580
1 Parent(s): 32b74e7

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -1
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 are 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()
 
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()