Cargo Features
RWuerstchen has no features set by default.
[dependencies]
rwuerstchen = { version = "0.4.0", features = ["flash", "accelerate", "cuda", "cudnn", "mkl", "nccl", "metal"] }anyhow.workspace = true
- flash
-
Enables flash-attn of candle-transformers ^0.8.2
- accelerate
-
Enables accelerate-src, accelerate of candle-core ^0.8.2, candle-nn ^0.8.2, and candle-transformers ^0.8.2
- cuda nccl?
-
Enables cuda of candle-core ^0.8.2, candle-nn ^0.8.2, and candle-transformers ^0.8.2
- cudnn
-
Enables cudnn of candle-core ^0.8.2
- mkl
-
Enables intel-mkl-src, mkl of candle-core ^0.8.2, candle-nn ^0.8.2, and candle-transformers ^0.8.2
- nccl = cuda
-
Enables half, nccl of cudarc ^0.9.14
- metal
-
Enables metal of candle-core ^0.8.2, candle-nn ^0.8.2, candle-transformers ^0.8.2, 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