浮点数例外 (核心已转储), Floating point exception (核心已转储)

问题描述

项目：https://github.com/RedAIGC/Flux-version-LayerDiffuse

报错：Floating point exception (核心已转储)

报错发生的位置：/path/diffusers/models/embeddings.py 的 1309 行，sample = self.linear_2(sample)

    def forward(self, sample, condition=None):
        if condition is not None:
            sample = sample + self.cond_proj(condition)
        sample = self.linear_1(sample)

        if self.act is not None:
            sample = self.act(sample)

        sample = self.linear_2(sample)
    
        if self.post_act is not None:
            sample = self.post_act(sample)
        return sample

但奇怪的是，报错不会显示问题出现在哪一行，而是直接中断，提示 Floating point exception (核心已转储)

解决方案

将 torch_dtype=torch.bfloat16 改为了 torch_dtype=torch.float32 后正常运行。