参考：https://zhuanlan.zhihu.com/p/644545784
1、修改 modules/api/models.py 中的 StableDiffusionTxt2ImgProcessingAPI 增加模型名称
StableDiffusionTxt2ImgProcessingAPI = PydanticModelGenerator( "StableDiffusionProcessingTxt2Img", StableDiffusionProcessingTxt2Img, [ {"key": "sampler_index", "type": str, "default": "Euler"}, {"key": "script_name", "type": str, "default": None}, {"key": "script_args", "type": list, "default": []}, {"key": "send_images", "type": bool, "default": True}, {"key": "save_images", "type": bool, "default": False}, {"key": "alwayson_scripts", "type": dict, "default": {}}, {"key": "model_name", "type": str, "default": None}, ] ).generate_model() 2、修改 modules/api/api.py 中 text2imgapi 代码：
...... from modules import sd_samplers, deepbooru, sd_hijack, images, scripts, ui, \ postprocessing, errors, restart, shared_items, sd_models from modules.api import models from modules.shared import opts, models_path ...... def text2imgapi(self, txt2imgreq: models.StableDiffusionTxt2ImgProcessingAPI): ...... with self.queue_lock: if txt2imgreq.model_name is not None: checkpoint_info = sd_models.CheckpointInfo(os.path.join(models_path, 'Stable-diffusion', txt2imgreq.model_name)) sd_models.reload_model_weights(info = checkpoint_info) with closing(StableDiffusionProcessingTxt2Img(sd_model=shared.sd_model, **args)) as p: ...... 3、修改 modules/processing.py 中的 StableDiffusionProcessingTxt2Img，增加模型名称接收
@dataclass(repr=False) class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing): enable_hr: bool = False denoising_strength: float = 0.75 firstphase_width: int = 0 firstphase_height: int = 0 hr_scale: float = 2.0 hr_upscaler: str = None hr_second_pass_steps: int = 0 hr_resize_x: int = 0 hr_resize_y: int = 0 hr_checkpoint_name: str = None hr_sampler_name: str = None hr_prompt: str = '' hr_negative_prompt: str = '' model_name: str = None
