memory_efficient_attention_forward

[RaykiDan]

Try to run the command "python3 run.py --encoder vitl --img-path assets/examples --outdir depth_vis"

But it showed error because of certain operator not found:

NotImplementedError: No operator found for memory_efficient_attention_forward with inputs:
query : shape=(1, 2073, 16, 64) (torch.float32)
key : shape=(1, 2073, 16, 64) (torch.float32)
value : shape=(1, 2073, 16, 64) (torch.float32)
attn_bias : <class 'NoneType'>
p : 0.0
[email protected] is not supported because:
device=cpu (supported: {'cuda'})
dtype=torch.float32 (supported: {torch.float16, torch.bfloat16})
cutlassF-pt is not supported because:

How to solve this? Thanks before