music
sander-wood commited on
Commit
8ae5c8d
·
verified ·
1 Parent(s): 5b90873

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -3
README.md CHANGED
@@ -56,9 +56,10 @@ To set up the MelodyT5 environment and install the necessary dependencies, follo
56
  ## Usage
57
 
58
  - `config.py`: Configuration settings for training and inference.
59
- - `generate.py`: Perform inference tasks (e.g., generation and conversion) using pre-trained models.
60
- - `train-cls.py`: Training script for classification models.
61
- - `train-gen.py`: Training script for generative models.
 
62
  - `utils.py`: Utility functions supporting model operations and data processing.
63
 
64
  ### Setting Up Inference Parameters
 
56
  ## Usage
57
 
58
  - `config.py`: Configuration settings for training and inference.
59
+ - `inference.py`: Perform inference tasks (e.g., generation and harmonization) using trained models.
60
+ - `prompt.txt`: Text file containing the prompt to specify a task and its input for the model.
61
+ - `random_model.py`: Script for creating random weights to initialize the model when using MelodyT5's patch-level encoder and decoder with shared weights.
62
+ - `train.py`: Script for training the MelodyT5 model.
63
  - `utils.py`: Utility functions supporting model operations and data processing.
64
 
65
  ### Setting Up Inference Parameters