diff --git a/src/main/java/com/ai/da/python/PythonService.java b/src/main/java/com/ai/da/python/PythonService.java index 9d123b9b..45d8a75a 100644 --- a/src/main/java/com/ai/da/python/PythonService.java +++ b/src/main/java/com/ai/da/python/PythonService.java @@ -555,7 +555,9 @@ public class PythonService { return 0; } else { long pinNum = printBoardElements.stream().filter(f -> f.getHasPin() == 1).count(); - if (designNum - pinNum < designNum/2) { + if (designNum - pinNum < 0){ + return RandomsUtil.randomSysFile(0L, (long) (pinNum/2 + 1)); + } else if (designNum - pinNum < designNum/2) { return RandomsUtil.randomSysFile(0L, designNum - pinNum + 1); } else { return RandomsUtil.randomSysFile(0L, (long) (designNum/2 + 1)); diff --git a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java index 3b08bcc4..a834f987 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -1284,6 +1284,8 @@ public class GenerateServiceImpl extends ServiceImpl i public String imageToSketchAsync(ImageToSketchDTO imageToSketchDTO, String collagePictureUrl, Long projectId) { Long accountId = UserContext.getUserHolder().getId(); log.info("imageToSketch parameter : {}", imageToSketchDTO); + // 目前没有用其他提取模型,暂时写死flux + imageToSketchDTO.setModelName("flux"); // 检查积分是否够本次扣除 // CreditsEventsEnum event = CreditsEventsEnum.IMAGE_TO_SKETCH;