| import torch | |
| # Path to your trained model checkpoint | |
| CHECKPOINT_PATH = "metalampnet_ckpt.pth" | |
| # Device to run inference on | |
| DEVICE = "cuda" if torch.cuda.is_available() else "cpu" | |
| # Sliding window size and prediction length | |
| WINDOW = 2048 | |
| PRED_SAMPLES = 512 | |
| # Output sample rate | |
| SR = 44100 | |