mirror of
https://github.com/huggingface/candle.git
synced 2025-06-18 11:37:11 +00:00
Remove some todos.
This commit is contained in:
@ -577,7 +577,6 @@ impl CpuStorage {
|
|||||||
layout_f: &Layout,
|
layout_f: &Layout,
|
||||||
) -> Result<Self> {
|
) -> Result<Self> {
|
||||||
// TODO: Support types that could be casted to a boolean.
|
// TODO: Support types that could be casted to a boolean.
|
||||||
// TODO: this should use the layout.
|
|
||||||
let pred = self.as_slice::<u32>()?;
|
let pred = self.as_slice::<u32>()?;
|
||||||
match (t, f) {
|
match (t, f) {
|
||||||
(Self::BF16(t), Self::BF16(f)) => {
|
(Self::BF16(t), Self::BF16(f)) => {
|
||||||
@ -615,7 +614,6 @@ impl CpuStorage {
|
|||||||
hidden_size: usize,
|
hidden_size: usize,
|
||||||
vocab_size: usize,
|
vocab_size: usize,
|
||||||
) -> Result<Self> {
|
) -> Result<Self> {
|
||||||
// TODO: this should use the layout.
|
|
||||||
let ids = self.as_slice::<u32>()?;
|
let ids = self.as_slice::<u32>()?;
|
||||||
map1!(vs, take_impl1, ids, layout, vocab_size, hidden_size)
|
map1!(vs, take_impl1, ids, layout, vocab_size, hidden_size)
|
||||||
}
|
}
|
||||||
|
Reference in New Issue
Block a user