TerraTunes / requirements.txt
Utsaha's picture
Create requirements.txt
55d6996 verified
raw
history blame
394 Bytes
torch==2.0.1 # Ensure that you have a compatible version for CUDA if using GPU
torchaudio==2.0.1 # Ensure compatibility with the PyTorch version
transformers==4.29.2 # Version of the Hugging Face Transformers library
scikit-learn==1.2.2 # For LabelEncoder
tf-keras==2.11.0 # TensorFlow Keras, version may vary based on your TensorFlow installation
pandas==2.0.1 # For handling dataframes