module 'torch' has no attribute 'float4_e2m1fn_x2'

#7
by Blueforcer - opened

im getting the error "module 'torch' has no attribute 'float4_e2m1fn_x2'" when running the FP4 model.
FP8 works. Do i need to update torch? Im using a RTX5090

image

you need to install ComfyKitchen to use FP4 https://github.com/Comfy-Org/comfy-kitchen

I have comfy-kitchen installed and FP4 still gives same error with RTX-5090

Lightricks org

To use NVFP4 on a 5090 you need a PyTorch version compiled with CUDA 13.0 or higher.

rluxemburg changed discussion status to closed

Sign up or log in to comment