# Lightning-based configuration for LSTM model
# Extends the base LSTM configuration with Lightning-specific settings

MODEL:
  ASR_VOCAB_SIZE: 91
  DFF: 128
  DROPOUT_RATE: 0.2
  D_MODEL: 256
  LOAD_TEXT_BRANCH_ONLY: false
  MAXLEN: 272
  NUM_BLOCKS: 2
  NUM_HEADS: 4
  OUTPUT_SIZE: 19
  PRETRAINED_PATH: null
  TYPE: Transformer
  USE_ASR: false
  VOCAB_SIZE: 77
  WITH_CONN: false

TRAIN:
  ACCUMULATE_GRAD_BATCHES: 1
  BATCH_SIZE: 128
  DEVICE: cuda
  EARLY_STOPPING_PATIENCE: 10
  EVAL_FREQ: 1
  LEARNING_RATE: 0.001
  NUM_EPOCHS: 100
  SAVE_DIR: results/transformer-text-only/tashkeela
  SAVE_FREQ: 30
  VAL_CHECK_INTERVAL: 1.0

INFERENCE:
  ASR_MODEL_NAME: sashat/whisper-medium-ClassicalAr
  BATCH_SIZE: 16
  BUFFER_SIZE: 25
  DEVICE: cuda
  FORCED_IDS: null
  MAX_LENGTH: 300
  MODEL_PATH: null
  OUTPUT_PATH: results/transformer-text-only_clartts.txt
  USE_ASR: false
  WINDOW_SIZE: 50

DATA:
  MAX_LENGTH: 270
  TEST_PATH: data/clartts/ClarTTS_test_Ref.txt
  TRAIN_PATH: data/tashkeela/train.txt
  VAL_PATH: data/tashkeela/val.txt

CONSTANTS_PATH: 'constants/'