@@ -55,20 +55,20 @@ public StabilityAiImageModel(StabilityAiApi stabilityAiApi, StabilityAiImageOpti
5555 private static StabilityAiApi .GenerateImageRequest getGenerateImageRequest (ImagePrompt stabilityAiImagePrompt ,
5656 StabilityAiImageOptions optionsToUse ) {
5757 return new StabilityAiApi .GenerateImageRequest .Builder ()
58- .withTextPrompts (stabilityAiImagePrompt .getInstructions ()
58+ .textPrompts (stabilityAiImagePrompt .getInstructions ()
5959 .stream ()
6060 .map (message -> new StabilityAiApi .GenerateImageRequest .TextPrompts (message .getText (),
6161 message .getWeight ()))
6262 .collect (Collectors .toList ()))
63- .withHeight (optionsToUse .getHeight ())
64- .withWidth (optionsToUse .getWidth ())
65- .withCfgScale (optionsToUse .getCfgScale ())
66- .withClipGuidancePreset (optionsToUse .getClipGuidancePreset ())
67- .withSampler (optionsToUse .getSampler ())
68- .withSamples (optionsToUse .getN ())
69- .withSeed (optionsToUse .getSeed ())
70- .withSteps (optionsToUse .getSteps ())
71- .withStylePreset (optionsToUse .getStylePreset ())
63+ .height (optionsToUse .getHeight ())
64+ .width (optionsToUse .getWidth ())
65+ .cfgScale (optionsToUse .getCfgScale ())
66+ .clipGuidancePreset (optionsToUse .getClipGuidancePreset ())
67+ .sampler (optionsToUse .getSampler ())
68+ .samples (optionsToUse .getN ())
69+ .seed (optionsToUse .getSeed ())
70+ .steps (optionsToUse .getSteps ())
71+ .stylePreset (optionsToUse .getStylePreset ())
7272 .build ();
7373 }
7474
@@ -124,32 +124,31 @@ StabilityAiImageOptions mergeOptions(ImageOptions runtimeOptions, StabilityAiIma
124124 }
125125 StabilityAiImageOptions .Builder builder = StabilityAiImageOptions .builder ()
126126 // Handle portable image options
127- .withModel (ModelOptionsUtils .mergeOption (runtimeOptions .getModel (), defaultOptions .getModel ()))
128- .withN (ModelOptionsUtils .mergeOption (runtimeOptions .getN (), defaultOptions .getN ()))
129- .withResponseFormat (ModelOptionsUtils .mergeOption (runtimeOptions .getResponseFormat (),
127+ .model (ModelOptionsUtils .mergeOption (runtimeOptions .getModel (), defaultOptions .getModel ()))
128+ .N (ModelOptionsUtils .mergeOption (runtimeOptions .getN (), defaultOptions .getN ()))
129+ .responseFormat (ModelOptionsUtils .mergeOption (runtimeOptions .getResponseFormat (),
130130 defaultOptions .getResponseFormat ()))
131- .withWidth (ModelOptionsUtils .mergeOption (runtimeOptions .getWidth (), defaultOptions .getWidth ()))
132- .withHeight (ModelOptionsUtils .mergeOption (runtimeOptions .getHeight (), defaultOptions .getHeight ()))
133- .withStylePreset (ModelOptionsUtils .mergeOption (runtimeOptions .getStyle (), defaultOptions .getStyle ()))
131+ .width (ModelOptionsUtils .mergeOption (runtimeOptions .getWidth (), defaultOptions .getWidth ()))
132+ .height (ModelOptionsUtils .mergeOption (runtimeOptions .getHeight (), defaultOptions .getHeight ()))
133+ .stylePreset (ModelOptionsUtils .mergeOption (runtimeOptions .getStyle (), defaultOptions .getStyle ()))
134134 // Always set the stability-specific defaults
135- .withCfgScale (defaultOptions .getCfgScale ())
136- .withClipGuidancePreset (defaultOptions .getClipGuidancePreset ())
137- .withSampler (defaultOptions .getSampler ())
138- .withSeed (defaultOptions .getSeed ())
139- .withSteps (defaultOptions .getSteps ())
140- .withStylePreset (defaultOptions .getStylePreset ());
135+ .cfgScale (defaultOptions .getCfgScale ())
136+ .clipGuidancePreset (defaultOptions .getClipGuidancePreset ())
137+ .sampler (defaultOptions .getSampler ())
138+ .seed (defaultOptions .getSeed ())
139+ .steps (defaultOptions .getSteps ())
140+ .stylePreset (defaultOptions .getStylePreset ());
141141 if (runtimeOptions instanceof StabilityAiImageOptions ) {
142142 StabilityAiImageOptions stabilityOptions = (StabilityAiImageOptions ) runtimeOptions ;
143143 // Handle Stability AI specific image options
144144 builder
145- .withCfgScale (
146- ModelOptionsUtils .mergeOption (stabilityOptions .getCfgScale (), defaultOptions .getCfgScale ()))
147- .withClipGuidancePreset (ModelOptionsUtils .mergeOption (stabilityOptions .getClipGuidancePreset (),
145+ .cfgScale (ModelOptionsUtils .mergeOption (stabilityOptions .getCfgScale (), defaultOptions .getCfgScale ()))
146+ .clipGuidancePreset (ModelOptionsUtils .mergeOption (stabilityOptions .getClipGuidancePreset (),
148147 defaultOptions .getClipGuidancePreset ()))
149- .withSampler (ModelOptionsUtils .mergeOption (stabilityOptions .getSampler (), defaultOptions .getSampler ()))
150- .withSeed (ModelOptionsUtils .mergeOption (stabilityOptions .getSeed (), defaultOptions .getSeed ()))
151- .withSteps (ModelOptionsUtils .mergeOption (stabilityOptions .getSteps (), defaultOptions .getSteps ()))
152- .withStylePreset (ModelOptionsUtils .mergeOption (stabilityOptions .getStylePreset (),
148+ .sampler (ModelOptionsUtils .mergeOption (stabilityOptions .getSampler (), defaultOptions .getSampler ()))
149+ .seed (ModelOptionsUtils .mergeOption (stabilityOptions .getSeed (), defaultOptions .getSeed ()))
150+ .steps (ModelOptionsUtils .mergeOption (stabilityOptions .getSteps (), defaultOptions .getSteps ()))
151+ .stylePreset (ModelOptionsUtils .mergeOption (stabilityOptions .getStylePreset (),
153152 defaultOptions .getStylePreset ()));
154153 }
155154
0 commit comments