huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 18:48:51 +00:00
Code Issues Packages Projects Releases Wiki Activity

Some CLIP fixes for stable diffusion. (#338)

Browse Source 
* Some CLIP fixes for stable diffusion.

* Add the avg-pool2d operation on cpu.
This commit is contained in:
Laurent Mazare
2023-08-07 19:31:45 +02:00
committed by GitHub
parent 2345b8ce3f
commit fc265d9dcf
7 changed files with 81 additions and 18 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
candle-core/src/storage.rs
View File

@ -268,11 +268,20 @@ impl Storage {
pub(crate) fn avg_pool2d(
&self,
_layout: &Layout,
_kernel_size: (usize, usize),
_stride: (usize, usize),
layout: &Layout,
kernel_size: (usize, usize),
stride: (usize, usize),
) -> Result<Self> {
todo!()
match self {
Storage::Cpu(storage) => {
let storage = storage.avg_pool2d(layout, kernel_size, stride)?;
Ok(Self::Cpu(storage))
}
Self::Cuda(storage) => {
let storage = storage.avg_pool2d(layout, kernel_size, stride)?;
Ok(Self::Cuda(storage))
}
}
}
pub(crate) fn upsample_nearest2d(