I use below to print the error message
except ImportError as e:
print(f"ImportError: {e}")
causal_dot_product_cuda = causal_dot_backward_cuda = None
it tells me ImportError: No module named 'fast_transformers.causal_product.causal_product_cuda'
So I think the issue is that Python is unable to load the causal_product_cuda.cu file in the causal_product directory under the fast_transformers package.
How can I resolve this?
I use below to print the error message except ImportError as e: print(f"ImportError: {e}") causal_dot_product_cuda = causal_dot_backward_cuda = None it tells me ImportError: No module named 'fast_transformers.causal_product.causal_product_cuda' So I think the issue is that Python is unable to load the causal_product_cuda.cu file in the causal_product directory under the fast_transformers package. How can I resolve this?