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

Make things contiguous.

Browse Source
This commit is contained in:
laurent
2023-09-19 14:04:56 +01:00
parent f2a648f313
commit 4114872aae
1 changed files with 4 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-transformers/src/models/wuerstchen/diffnext.rs
View File

@ -37,7 +37,7 @@ impl ResBlockStageB {
let xs = xs.apply(&self.depthwise)?.apply(&self.norm)?;
let xs = match x_skip {
None => xs.clone(),
Some(x_skip) => Tensor::cat(&[&xs, x_skip], 1)?,
Some(x_skip) => Tensor::cat(&[&xs, x_skip], 1)?.contiguous()?,
};
let xs = xs
.permute((0, 2, 3, 1))?
@ -352,7 +352,9 @@ impl WDiffNeXt {
None
};
let skip = match (skip, effnet_c.as_ref()) {
(Some(skip), Some(effnet_c)) => Some(Tensor::cat(&[skip, effnet_c], 1)?),
(Some(skip), Some(effnet_c)) => {
Some(Tensor::cat(&[skip, effnet_c], 1)?.contiguous()?)
}
(None, Some(skip)) | (Some(skip), None) => Some(skip.clone()),
(None, None) => None,
};