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