修改 Stable Diffusion 使 api 接口增加模型参数

 参考：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/processing.py 中的 StableDiffusionProcessingTxt2Img，增加模型名称接收

@dataclass(repr=False)
class StableDiffusionProcessingTxt2Img(StableDiffusionProcessing):enable_hr: bool = Falsedenoising_strength: float = 0.75firstphase_width: int = 0firstphase_height: int = 0hr_scale: float = 2.0hr_upscaler: str = Nonehr_second_pass_steps: int = 0hr_resize_x: int = 0hr_resize_y: int = 0hr_checkpoint_name: str = Nonehr_sampler_name: str = Nonehr_prompt: str = ''hr_negative_prompt: str = ''model_name: str = None

 3.修改 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):......model_name = txt2imgreq.model_nameif model_name is None:raise HTTPException(status_code=404, detail="model_name not found")......with self.queue_lock:checkpoint_info = sd_models.CheckpointInfo(os.path.join(models_path, 'Stable-diffusion', model_name))sd_models.reload_model_weights(info=checkpoint_info)with closing(StableDiffusionProcessingTxt2Img(sd_model=shared.sd_model, **args)) as p:......