ClipCap-Chinese icon indicating copy to clipboard operation
ClipCap-Chinese copied to clipboard

cpu上 预测,报错怎么回事呀,GPU上 不会,RuntimeError: invalid multinomial distribution (sum of probabilities <= 0)

Open AI-Hao07 opened this issue 1 year ago • 0 comments

Traceback (most recent call last): File "predict.py", line 227, in main(args) File "predict.py", line 152, in main captions = generate(model, clip_embeds, tokenizer, args) File "predict.py", line 88, in generate next_token_ids = torch.multinomial(F.softmax(filtered_logits, dim=-1), num_samples=1).squeeze(1).tolist() RuntimeError: invalid multinomial distribution (sum of probabilities <= 0)

AI-Hao07 avatar Jul 07 '23 15:07 AI-Hao07