huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-18 03:28:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add some display tests + bugfixes.

Browse Source
This commit is contained in:
laurent
2023-06-27 21:37:28 +01:00
parent 8c81a70170
commit b0f5f2d22d
3 changed files with 102 additions and 14 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/display.rs
View File

@ -186,8 +186,8 @@ trait TensorFormatter {
.and_then(|t| t.to_vec1::<Self::Elem>())
{
for v in vs.into_iter() {
self.fmt(v, max_w, f)?;
write!(f, ", ")?;
self.fmt(v, max_w, f)?;
}
}
}
@ -257,7 +257,7 @@ struct FloatFormatter<S: WithDType> {
impl<S> FloatFormatter<S>
where
S: WithDType + num_traits::Float,
S: WithDType + num_traits::Float + std::fmt::Display,
{
fn new(t: &Tensor, po: &PrinterOptions) -> Result<Self> {
let mut int_mode = true;
@ -275,6 +275,7 @@ where
let mut nonzero_finite_min = S::max_value();
let mut nonzero_finite_max = S::min_value();
for &v in values.iter() {
let v = v.abs();
if v < nonzero_finite_min {
nonzero_finite_min = v
}
@ -289,7 +290,6 @@ where
break;
}
}
if let Some(v1) = S::from(1000.) {
if let Some(v2) = S::from(1e8) {
if let Some(v3) = S::from(1e-4) {