running FreeU on SM_52 cards

[lulu1315]

hello. i'm trying to use freeU with SD1.5 on a TitanX 12Gb card and i get this message :
RuntimeError: cuFFT doesn't support signals of half type with compute capability less than SM_53, but the device containing input half tensor only has SM_52

the strange thing is that it runs perfectly when used with SDXL so i was wondering ..
(i'm using diffusers)

• diffusers version: 0.27.0
• Platform: Linux-5.15.0-94-generic-x86_64-with-glibc2.29
• Python version: 3.8.0
• PyTorch version (GPU?): 2.2.1+cu121 (True)
• Huggingface_hub version: 0.21.4
• Transformers version: 4.38.2
• Accelerate version: 0.27.2
• xFormers version: not installed

[w990612]

错误的根本原因是你的 GPU 计算能力为 SM_52，不支持 cuFFT 中的半精度浮点数操作。
你可以通过将数据类型改为 float32 或将计算迁移到更高计算能力的设备上来解决此问题。