Forráskód Böngészése

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

cherishsince 1 éve
szülő
commit
bdbe2633e1

+ 4 - 3
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();
         StringBuilder params = new StringBuilder();
         //  --ar 来设置尺寸
         //  --ar 来设置尺寸
         params.append(String.format(" --ar %s:%s ", width, height));
         params.append(String.format(" --ar %s:%s ", width, height));
-        // --v 版本
-        params.append(String.format(" --v %s ", version));
         // --niji 模型
         // --niji 模型
         if (MidjourneyApi.ModelEnum.NIJI == model) {
         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();
         return params.toString();
     }
     }