mirror of
https://github.com/huggingface/candle.git
synced 2025-06-16 18:48:51 +00:00
Starcoder fix (#264)
* Bugfix for starcoder. * Get some proper code generation. * Slightly simpler softmax.
This commit is contained in:
@ -705,7 +705,8 @@ impl<'a, I: IntDType> Map1 for IndexSelect<'a, I> {
|
||||
expected: 1,
|
||||
got: d.len(),
|
||||
shape: self.ids_l.shape().clone(),
|
||||
})?,
|
||||
}
|
||||
.bt())?,
|
||||
};
|
||||
let stride_ids = self.ids_l.stride()[0];
|
||||
let mut dst_dims = layout.dims().to_vec();
|
||||
|
Reference in New Issue
Block a user