search

idiap / fast-transformers

Pytorch library for fast transformer implementations
1.65k stars 179 forks source link

Quick start raise a ModuleNotFoundError #99

Closed CaoYiqingT closed 3 years ago

CaoYiqingT commented 3 years ago

I have successfully prepared the required environment using "pip install --user pytorch-fast-transformers"

When I tried to run the quick start code, problem raised as follow: Traceback (most recent call last): File "quick_start.py", line 2, in from fast_transformers.builders import TransformerEncoderBuilder File "/home/caoyq/fast-transformers-master/fast_transformers/builders/init.py", line 42, in from ..attention import \ File "/home/caoyq/fast-transformers-master/fast_transformers/attention/init.py", line 13, in from .causal_linear_attention import CausalLinearAttention File "/home/caoyq/fast-transformers-master/fast_transformers/attention/causal_linear_attention.py", line 15, in from ..causal_product import causal_dot_product File "/home/caoyq/fast-transformers-master/fast_transformers/causal_product/init.py", line 9, in from .causal_product_cpu import causal_dot_product as causal_dot_product_cpu, \ ModuleNotFoundError: No module named 'fast_transformers.causal_product.causal_product_cpu'

lipmem commented 3 years ago

Which torch version are you using? For me it works with torch==1.8.1

angeloskath commented 3 years ago

Hi,

I will close the issue due to inactivity, feel free to reopen it if needed.

Cheers, Angelos