fix: validate safetensors data offsets (#1754)

This commit is contained in:
leejet authored and GitHub committed 2026-07-06 22:21:08 +08:00
1 parent 4fcc6fef10
commit e22272ee63
2 files changed
+12 -7

No files matched your search

+4 -4
View File
@@ -1897,10 +1897,10 @@ public:
}
sd::Tensor<float> operator()(double sigma_a, double sigma_b) {
double a = clamp(std::min(sigma_a, sigma_b));
double b = clamp(std::max(sigma_a, sigma_b));
auto dW = w(b) - w(a);
float span = static_cast<float>(std::max(std::abs(sigma_b - sigma_a), 1e-12));
double a = clamp(std::min(sigma_a, sigma_b));
double b = clamp(std::max(sigma_a, sigma_b));
auto dW = w(b) - w(a);
float span = static_cast<float>(std::max(std::abs(sigma_b - sigma_a), 1e-12));
return dW * (1.0f / std::sqrt(span));
}