Fix requirements (#295)

* Fix requirements

* Fix requirements

environment_cpu.yml

@@ -31,5 +31,6 @@ dependencies:
       - pre-commit
       - pytorch_lightning
       - sounddevice
+      - transformers>=4  # See requirements.txt
       - wavio >=0.0.5
       - -e .

environment_gpu.yml

@@ -33,5 +33,6 @@ dependencies:
       - pre-commit
       - pytorch_lightning
       - sounddevice
+      - transformers>=4  # See requirements.txt
       - wavio >=0.0.5
       - -e .

nam/_version.py

@@ -1 +1 @@
-__version__ = "0.7.0"
+__version__ = "0.7.1"

requirements.txt

@@ -18,6 +18,9 @@ pytorch_lightning
 scipy
 sounddevice
 torch
+# Not required, but if you have it, it needs to be recent enough so I'm adding
+# it.
+transformers>=4
 tqdm
 wavio
 wheel

setup.py

@@ -5,6 +5,17 @@
 from distutils.util import convert_path
 from setuptools import setup, find_packages
 
+def get_additional_requirements():
+    # Issue 294
+    try:
+        import transformers
+        # This may not be unnecessarily straict a requirement, but I'd rather
+        # fix this promptly than leave a chance that it wouldn't be fixed 
+        # properly.
+        return ["transformers>=4"]
+    except ModuleNotFoundError:
+        return []
+
 main_ns = {}
 ver_path = convert_path("nam/_version.py")
 with open(ver_path) as ver_file:
@@ -14,6 +25,8 @@
     "auraloss==0.3.0",
     "matplotlib",
     "numpy",
+    "onnx",
+    "onnxruntime",
     "pydantic",
     "pytorch_lightning",
     "scipy",
@@ -23,6 +36,7 @@
     "tqdm",
     "wavio>=0.0.5",  # Breaking change with older versions
 ]
+requirements.extend(get_additional_requirements())
 
 setup(
     name="neural-amp-modeler",