pub use DType and Tensor
This commit is contained in:
+1
-1
@@ -6,4 +6,4 @@ pub mod position_embed;
|
||||
pub mod tokenizer;
|
||||
pub mod utils;
|
||||
|
||||
pub use candle_core::Device;
|
||||
pub use candle_core::{DType, Device, Tensor};
|
||||
|
||||
Reference in New Issue
Block a user