From 6f08e33727e60051f587bc479ada25d8e97027a7 Mon Sep 17 00:00:00 2001
From: Jade Copet <jadecopet@fb.com>
Date: Tue, 21 Nov 2023 15:06:29 +0000
Subject: [PATCH 1/2] Fix linter in CI

---
 audiocraft/utils/cache.py | 1 +
 1 file changed, 1 insertion(+)

diff --git a/audiocraft/utils/cache.py b/audiocraft/utils/cache.py
index f7f82064..6ba017a7 100644
--- 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:

From d7e4acca093bdef7b6c67fa18f55a0f24481abd4 Mon Sep 17 00:00:00 2001
From: Jade Copet <jadecopet@fb.com>
Date: Tue, 21 Nov 2023 08:53:25 -0800
Subject: [PATCH 2/2] Set pytorch version to 2.1.0

---
 requirements.txt | 4 ++--
 1 file changed, 2 insertions(+), 2 deletions(-)

diff --git a/requirements.txt b/requirements.txt
index e44fe159..a6fa5809 100644
--- 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
\ No newline at end of file
+protobuf