chenyu f96fc6e9d4 fix gpt2 with empty prompt take 2 (#3102) ...

logits would be empty so need to replace that with ones before sampling, also cannot reshape with -1 when there's 0 in other axes

2024-01-12 14:46:36 -05:00

9.3 KiB

Raw Blame History

View Raw