mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 11:37:11 +00:00
d34039e352
* Start adding a stable-diffusion example. * Proper computation of the causal mask. * Add the chunk operation. * Work in progress: port the attention module. * Add some dummy modules for conv2d and group-norm, get the attention module to compile. * Re-enable the 2d convolution. * Add the embeddings module. * Add the resnet module. * Add the unet blocks. * Add the unet. * And add the variational auto-encoder. * Use the pad function from utils.
18 lines
288 B
Rust
18 lines
288 B
Rust
use candle::{Result, Tensor};
|
|
|
|
pub fn sigmoid(_: &Tensor) -> Result<Tensor> {
|
|
todo!()
|
|
}
|
|
|
|
pub fn avg_pool2d(_: &Tensor) -> Result<Tensor> {
|
|
todo!()
|
|
}
|
|
|
|
pub fn pad(_: &Tensor) -> Result<Tensor> {
|
|
todo!()
|
|
}
|
|
|
|
pub fn upsample_nearest2d(_: &Tensor) -> Result<Tensor> {
|
|
todo!()
|
|
}
|