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

Bugfixes. (#97)

Browse Source
This commit is contained in:
Laurent Mazare
2023-07-06 23:26:11 +01:00
committed by GitHub
parent a3f3b93d16
commit 2b8e8c9f14
2 changed files with 5 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-examples/examples/falcon/model.rs
View File

@ -444,7 +444,7 @@ impl FalconAttention {
.reshape((b_sz, self.num_heads, q_len, head_dim))?
.transpose(1, 2)?
.reshape((b_sz, q_len, self.num_heads * head_dim))?;
let attn_output = self.attn_output.forward(&attn_output)?;
let attn_output = self.dense.forward(&attn_output)?;
Ok(attn_output)
}
}