text: Title: Dirichlet Likelihood TFT model
  Body: Hello,
  I'm tring to use DirichletLikelihood in TFT model, but I got this issue. How can I fix it?
  ```python
  `ValueError: Expected value argument (Tensor of shape (32, 12, 1)) to be within the support (Simplex()) of the distribution Dirichlet(concentration: torch.Size([12, 1])), but found invalid values:
  tensor([[[0.5691],
           [0.8910],
           [0.6211],
           [0.0000],
           [0.3685],
           [0.3022],
           [0.5626],
           [0.3967],
           [0.6753],
           [0.6197],
           [0.4106],
           [0.6041]],
  ...
          [[0.0000],
           [0.0000],
           [0.0000],
           [0.0000],
           [0.0000],
           [0.0000],
           [0.4241],
           [0.3950],
           [0.3232],
           [0.2716],
           [0.3947],
           [0.2650]]])`
  ```
  Thanks