text: from typing import Any, Dict
  
  from .forward_lstm import ForwardLSTM
  
  
  class AWDLSTM(ForwardLSTM):
      def __init__(self, *args: Any, **kwargs: Any) -> None:
          kwargs.setdefault("rnn_name", "rnns")
          super().__init__(*args, **kwargs)
  
      @staticmethod
      def param_names(layer: int, rnn_name: str, **kwargs) -> Dict[str, str]:
          return {
              "weight_hh": f"{rnn_name}.{layer}.module.weight_hh_l0_raw",
              "weight_ih": f"{rnn_name}.{layer}.module.weight_ih_l0",
              "bias_hh": f"{rnn_name}.{layer}.module.bias_hh_l0",
              "bias_ih": f"{rnn_name}.{layer}.module.bias_ih_l0",
          }
  