diff --git a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java index d881e5b..9be2f56 100644 --- a/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java +++ b/bnyer-services/bnyer-img/src/main/java/com/bnyer/img/service/impl/StableDiffusionServiceImpl.java @@ -41,20 +41,22 @@ public class StableDiffusionServiceImpl implements StableDiffusionService { @Override public TextToImgVo textToImg(TextToImgDto param) { try{ - String prompt = ""; + //String prompt = ""; //判断prompt是否包含中文,中文则翻译,否则跳过 - if(TranslateUtils.isContainChinese(param.getPrompt())){ - //调用翻译api - //prompt = - }else{ - prompt = param.getPrompt(); - } +// if(TranslateUtils.isContainChinese(param.getPrompt())){ +// //调用翻译api +// //prompt = +// prompt = param.getPrompt(); +// }else{ +// prompt = param.getPrompt(); +// } //TODO 根据选择的风格来选择模型 Map map = new HashMap<>(); map.put("width",param.getWidth() == null ? 512 : param.getWidth()); map.put("height",param.getHeight() == null ? 512 : param.getHeight()); - map.put("prompt", prompt); + //map.put("prompt", prompt); + map.put("prompt", param.getPrompt()); map.put("seed",-1); map.put("batch_size",1); map.put("cfg_scale",7); @@ -66,7 +68,7 @@ public class StableDiffusionServiceImpl implements StableDiffusionService { map.put("steps",25); map.put("negative_prompt","nsfw"); log.info("请求stable_diffusion请求体为:【{}】", JSON.toJSONString(map)); - JSONObject jsonObject = restTemplate.postForObject("http://localhost:7860/sdapi/v1/txt2img", map, JSONObject.class); + JSONObject jsonObject = restTemplate.postForObject("http://124.220.2.237:7860/sdapi/v1/txt2img", map, JSONObject.class); log.info("请求stable_diffusion响应体的为:【{}】", JSON.toJSONString(jsonObject)); TextToImgVo img = new TextToImgVo(); if(jsonObject != null && jsonObject.getJSONArray("images").size() > 0){