diff --git a/songgeneration/codeclm/models/llama/tokenization_llama.py b/songgeneration/codeclm/models/llama/tokenization_llama.py
index bf105b7..d8a2a60 100644
--- a/songgeneration/codeclm/models/llama/tokenization_llama.py
+++ b/songgeneration/codeclm/models/llama/tokenization_llama.py
@@ -26,7 +26,11 @@ from typing import TYPE_CHECKING, Any, Dict, List, Optional, Tuple
 import sentencepiece as spm
 
 from transformers.convert_slow_tokenizer import import_protobuf
-from transformers.tokenization_utils import AddedToken, PreTrainedTokenizer
+try:
+    from transformers.tokenization_utils import AddedToken, PreTrainedTokenizer
+except ImportError:
+    from transformers.tokenization_python import PreTrainedTokenizer
+    from transformers.tokenization_utils_base import AddedToken
 from transformers.utils import logging
 
 
diff --git a/songgeneration/requirements.txt b/songgeneration/requirements.txt
index 306b588..7b9070b 100644
--- a/songgeneration/requirements.txt
+++ b/songgeneration/requirements.txt
@@ -1,20 +1,20 @@
 alias-free-torch>=0.0.6
 diffusers>=0.37.1,<0.38
 einops>=0.8.1
-einops-exts==0.0.4
+einops-exts>=0.0.4
 huggingface-hub>=0.34.0,<2.0
 julius>=0.2.7
-k-diffusion==0.1.1
+k-diffusion>=0.1.1,<0.2
 lameenc>=1.8.1
 librosa>=0.11.0
-omegaconf==2.2.0
+omegaconf>=2.2.0,<3.0
 openunmix>=1.3.0
 packaging>=24.0
 pydantic>=2.0
 PyYAML>=6.0
 safetensors>=0.4.3
-torch==2.6.0
-torchaudio==2.6.0
+torch>=2.6.0,<2.12
+torchaudio>=2.6.0,<2.12
 tqdm>=4.66.0
 transformers>=4.37.2,<6.0
 vector-quantize-pytorch>=1.22.17
diff --git a/songgeneration/requirements_nodeps.txt b/songgeneration/requirements_nodeps.txt
index 5fec19a..6505eac 100644
--- a/songgeneration/requirements_nodeps.txt
+++ b/songgeneration/requirements_nodeps.txt
@@ -1,2 +1,2 @@
-antlr4-python3-runtime==4.8
-omegaconf==2.2.0
+antlr4-python3-runtime>=4.9,<4.10
+omegaconf>=2.2.0,<3.0