Starcoder fix (#264)

* Bugfix for starcoder.

* Get some proper code generation.

* Slightly simpler softmax.
This commit is contained in:
Laurent Mazare
2023-07-28 11:17:49 +01:00
committed by GitHub
parent 6a54ca115e
commit 3e89df938c
3 changed files with 20 additions and 7 deletions

View File

@ -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();