Cargo Features
RWuerstchen has no features set by default.
[dependencies]
rwuerstchen = { version = "0.3.2", features = ["flash", "accelerate", "cuda", "cudnn", "mkl", "nccl", "metal"] }- flash
-
Enables flash-attn of candle-transformers ^0.6.0
- accelerate
-
Enables accelerate-src, accelerate of candle-core ^0.6.0, candle-nn ^0.6.0, and candle-transformers ^0.6.0
- cuda nccl?
-
Enables cuda of candle-core ^0.6.0, candle-nn ^0.6.0, and candle-transformers ^0.6.0
- cudnn
-
Enables cudnn of candle-core ^0.6.0
- mkl
-
Enables intel-mkl-src, mkl of candle-core ^0.6.0, candle-nn ^0.6.0, and candle-transformers ^0.6.0
- nccl = cuda
-
Enables half, nccl of cudarc ^0.9.14
- metal
-
Enables metal of candle-core ^0.6.0, candle-nn ^0.6.0, candle-transformers ^0.6.0, and kalosm-common
Features from optional dependencies
In crates that don't use the dep: syntax, optional dependencies automatically become Cargo features. These features may have been created by mistake, and this functionality may be removed in the future.
- cudarc nccl?
-
Enables cudarc ^0.9.14