# From #289 # The issue The composite model returns logits, whereas the old one returned probs. # The correction to be applied Remove conversion from probs to logits.