【优化】调整 niji 模型，版本好参数

cherishsince
1 parent 4c89342d
Showing 1 changed file with 4 additions and 3 deletions
yudao-module-ai/yudao-module-ai-biz/src/main/java/cn/iocoder/yudao/module/ai/service/image/AiImageServiceImpl.java
@@ -300,11 +300,12 @@ public class AiImageServiceImpl implements AiImageService {
         StringBuilder params = new StringBuilder();
         //  --ar 来设置尺寸
         params.append(String.format(" --ar %s:%s ", width, height));
-        // --v 版本
-        params.append(String.format(" --v %s ", version));
         // --niji 模型
         if (MidjourneyApi.ModelEnum.NIJI == model) {
-            params.append(" --niji ");
+            params.append(String.format(" --niji %s ", version));
+        } else {
+            // --v 版本
+            params.append(String.format(" --v %s ", version));
         }
         return params.toString();
     }