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 dfe4dd30..97c495e5 100644 --- a/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java +++ b/src/main/java/com/ai/da/service/impl/GenerateServiceImpl.java @@ -2042,7 +2042,9 @@ public class GenerateServiceImpl extends ServiceImpl i && func.equals(CreditsEventsEnum.POSE_TRANSFORMATION.getValue())){ List poseTransformations = poseTransformationMapper.selectList( new QueryWrapper().eq("unique_id", taskId)); - if (!poseTransformations.isEmpty() && poseTransformations.get(0).getModelName().equals("wx")){ + if (!poseTransformations.isEmpty() + && !StringUtil.isNullOrEmpty(poseTransformations.get(0).getModelName()) + && poseTransformations.get(0).getModelName().equals("wx")){ return "wx"; }else { return "local";