upstream-archive/stable-diffusion.cpp
1
0
Fork 0
mirror of https://github.com/leejet/stable-diffusion.cpp.git synced 2026-10-03 03:17:51 -05:00
Code Issues Packages Projects Releases Wiki Activity

fix: validate safetensors data offsets (#1754)

Browse Source
This commit is contained in:
leejet
2026-07-06 22:21:08 +08:00
committed by GitHub
parent 4fcc6fef10
commit e22272ee63
2 changed files with 12 additions and 7 deletions
Download Patch File Download Diff File Expand all files Collapse all files
src/runtime/denoiser.hpp
+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));
}