Merge branch 'main' into uses_trajectories

requirements.txt

-numpy
+numpy>=1.24.3
 torch>=1.6
 torchaudio>=0.9.0
-tqdm
+tqdm>=4.65.0
+tensorboard>=2.13.0
+Pillow==9.2.0
\ No newline at end of file

src/diffwave/preprocess.py

@@ -23,6 +23,12 @@ from concurrent.futures import ProcessPoolExecutor
 from glob import glob
 from tqdm import tqdm
 
+# Needed to import from src/ in general
+import os
+import sys
+SOURCE_DIR = os.path.abspath(os.path.join(os.path.dirname( __file__ ), '../..', 'src'))
+sys.path.append(SOURCE_DIR)
+
 from diffwave.params import params