No module named 'triton', python==3.11.3

[jackyin68]

First step
python pipelines/sample.py --config configs/sample.yaml

Error

A matching Triton is not available, some optimizations will not be enabled.
Error caught was: No module named 'triton'

How to fix it, Or someone share the wheel for us.
Thanks in advance!

[maxin-cn]

First step python pipelines/sample.py --config configs/sample.yaml

Error

A matching Triton is not available, some optimizations will not be enabled. Error caught was: No module named 'triton'

How to fix it, Or someone share the wheel for us. Thanks in advance!

@jackyin68 Hi, is this mistake caused by xformers? Maybe you can comment this code (

LaVie/base/pipelines/sample.py

Line 65 in 0b925ff

videogen_pipeline.enable_xformers_memory_efficient_attention()

).

[jackyin68]

Many thanks to your kindly reply!