Label Knight

I meet the same question, I have changed the config and it show right if only use matplotlib, but pandas profiling doesn't work @sbrugman

我今天又看了下，我测试的是14B，如果是72B full的话，即使是80G的显卡也撑不过model.float()

我也遇到同样的问题

> torch.where call the torch.nonzer o function