• zhangp365's avatar
    fixed a dtype bfloat16 bug in torch_utils.py (#10125) · 188bca30
    zhangp365 authored
    
    
    * fixed a dtype bfloat16 bug in torch_utils.py
    
    when generating 1024*1024 image with bfloat16 dtype, there is an exception:
      File "/opt/conda/lib/python3.10/site-packages/diffusers/utils/torch_utils.py", line 107, in fourier_filter
        x_freq = fftn(x, dim=(-2, -1))
    RuntimeError: Unsupported dtype BFloat16
    
    * remove whitespace in torch_utils.py
    
    * Update src/diffusers/utils/torch_utils.py
    
    * Update torch_utils.py
    
    ---------
    Co-authored-by: default avatarhlky <hlky@hlky.ac>
    188bca30
torch_utils.py 6.2 KB