Allow disabling xFormers via environment variable (#180)
Allow disabling the use of xFormers (for inference) by simply setting the XFORMERS_DISABLED environment variable
Showing
- dinov2/layers/attention.py 14 additions, 5 deletionsdinov2/layers/attention.py
- dinov2/layers/block.py 14 additions, 5 deletionsdinov2/layers/block.py
- dinov2/layers/swiglu_ffn.py 12 additions, 2 deletionsdinov2/layers/swiglu_ffn.py
- dinov2/train/ssl_meta_arch.py 2 additions, 4 deletionsdinov2/train/ssl_meta_arch.py
Loading
Please register or sign in to comment