flask_default_config.cfg

FS_HZ = 16000
N_FFT = 2048
HOP_LENGTH = 512
MAX_SOUND_DURATION_S = 60

SPECTROGRAMS_UPSAMPLING_FACTOR = 1
DEFAULT_COLORMAP = 'viridis'

WARMUP_MODELS = True
USE_LOCAL_CONDITIONING = False
SAMPLING_TOP_K = 0
USE_PREDICTIVE_SAMPLING = True
SAMPLING_TOP_P = 0.99

PYTORCH_DEVICE = 'cuda'

PORT = 8000
MAX_CONTENT_LENGTH=10 * 1024 * 1024  # limit size of data uploaded to the server

# default saved models paths (useful e.g. for Docker-based deployment)
VQVAE_PATH = 'scripted_models/vqvae.pt'
TRANSFORMER_TOP_PATH = 'scripted_models/transformer_top.pt'
TRANSFORMER_BOTTOM_PATH = 'scripted_models/transformer_bottom.pt'