fix: avoid writable mmap for read-only weights (#1698)

This commit is contained in:
leejet authored and GitHub committed 2026-06-23 00:39:31 +08:00
1 parent 41f7acbfb0
commit f440ad9c29
6 files changed
+24 -22

No files matched your search

+5 -5
View File
@@ -3,9 +3,9 @@
#include <algorithm>
#include <cmath>
#include <cstdlib>
#include <optional>
#include <string>
#include <utility>
#include <optional>
#include "core/util.h"
@@ -127,7 +127,7 @@ namespace sd::guidance {
std::vector<float> parse_guidance_schedule(const char* extra_sample_args) {
std::vector<float> guidance_schedule;
std::string guidance_schedule_str = "";
for (const auto& [key, value] : parse_key_value_args(extra_sample_args, "extra sample arg")) {
for (const auto& [key, value] : parse_key_value_args(extra_sample_args, "extra sample arg")) {
float parsed = 0.0f;
if (key == "guidance_schedule") {
guidance_schedule_str = value;
@@ -148,9 +148,9 @@ namespace sd::guidance {
GuiderOutput ClassifierFreeGuidance::forward(const GuidanceInput& input,
GuiderOutput previous,
std::optional<float> scale_override) const {
std::optional<float> scale_override) const {
(void)previous;
float guidance_scale = scale_override.value_or(guidance_scale_);
float guidance_scale = scale_override.value_or(guidance_scale_);
GuiderOutput output;
if (!has_tensor(input.pred_cond)) {
@@ -210,7 +210,7 @@ namespace sd::guidance {
GuiderOutput previous,
std::optional<float> scale_override) const {
(void)previous;
float guidance_scale = scale_override.value_or(guidance_scale_);
float guidance_scale = scale_override.value_or(guidance_scale_);
GuiderOutput output;
if (!has_tensor(input.pred_cond)) {
+2 -2
View File
@@ -3,8 +3,8 @@
#include <cstddef>
#include <functional>
#include <vector>
#include <optional>
#include <vector>
#include "core/tensor.hpp"
@@ -42,7 +42,7 @@ namespace sd::guidance {
class BaseGuidance {
public:
virtual ~BaseGuidance() = default;
virtual ~BaseGuidance() = default;
virtual GuiderOutput forward(const GuidanceInput& input,
GuiderOutput previous,
std::optional<float> scale_override = std::nullopt) const = 0;