Merge pull request #352 from facebookresearch/fix_linter

Fix linter in CI + set PyTorch version
facebookresearch · Nov 21, 2023 · aaead2d · aaead2d
2 parents 1c570cd + d7e4acc
commit aaead2d
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 2 deletions.
diff --git a/audiocraft/utils/cache.py b/audiocraft/utils/cache.py
@@ -287,6 +287,7 @@ def _load_one(self, index: int):
                     if isinstance(part[0], torch.Tensor):
                         out.append(torch.stack(part))
                     else:
+                        assert isinstance(part, torch.Tensor)
                         out.append(part)
                 return out
         except Exception:

diff --git a/requirements.txt b/requirements.txt
@@ -9,7 +9,7 @@ num2words
 numpy
 sentencepiece
 spacy==3.5.2
-torch>=2.0.0
+torch==2.1.0
 torchaudio>=2.0.0
 huggingface_hub
 tqdm
@@ -20,4 +20,4 @@ librosa
 gradio
 torchmetrics
 encodec
-protobuf
+protobuf