Dirichlet Likelihood TFT model

Hello, I'm tring to use DirichletLikelihood in TFT model, but I got this issue. How can I fix it?

`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