added support for stability.sd3-5-large

techthoughts2 · Jan 5, 2025 · c3fc3f9 · c3fc3f9
1 parent eba68aa
commit c3fc3f9
Show file tree

Hide file tree

Showing 19 changed files with 87 additions and 17 deletions.
diff --git a/docs/CHANGELOG.md b/docs/CHANGELOG.md
@@ -5,7 +5,7 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.2.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
-## [0.50.0] - **BREAKING CHANGES**
+## [0.51.0] - **BREAKING CHANGES**
 
 - Module changes:
     - Added two new properties to all model documentation objects:
@@ -24,6 +24,8 @@ and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0
             - `CfgScale` parameter maximum adjusted from 10 to 9.9
     - Meta
         - Removed `meta.llama2-13b-chat-v1` and `meta.llama2-70b-chat-v1` from all functions as Bedrock has EOL these two models. - ***Breaking Change***
+    - stability.ai
+        - Updated `Invoke-StabilityAIImageModel` to support new model: `stability.sd3-5-large-v1:0`
 
 ## [0.33.0] - **BREAKING CHANGES**
 

diff --git a/docs/pwshBedrock.md b/docs/pwshBedrock.md
@@ -2,7 +2,7 @@
 Module Name: pwshBedrock
 Module Guid: b4f9e4dc-0229-44ef-99a1-08be4c5e81f2
 Download Help Link: NA
-Help Version: 0.50.0
+Help Version: 0.51.0
 Locale: en-US
 ---
 

diff --git a/src/Tests/Integration/ConverseAPI.Tests.ps1 b/src/Tests/Integration/ConverseAPI.Tests.ps1
@@ -53,6 +53,7 @@ InModuleScope 'pwshBedrock' {
                 # 'stability.stable-image-ultra-v1:0', # *note: not supported by Converse API
                 # 'stability.stable-image-core-v1:0' # *note: not supported by Converse API
                 # 'stability.sd3-large-v1:0' # *note: not supported by Converse API
+                # 'stability.sd3-5-large-v1:0' # *note: not supported by Converse API
             )
         }
         BeforeAll {

diff --git a/src/Tests/Integration/StabilityAI.Tests.ps1 b/src/Tests/Integration/StabilityAI.Tests.ps1
@@ -162,7 +162,7 @@ InModuleScope 'pwshBedrock' {
                 $eval | Should -Not -BeNullOrEmpty
             } #it
 
-            It 'should return an image when using image-to-image for <_.ModelId>' -Foreach ($script:stabilityAIModelInfo | Where-Object { $_.ModelID -eq 'stability.sd3-large-v1:0' }) {
+            It 'should return an image when using image-to-image for <_.ModelId>' -Foreach ($script:stabilityAIModelInfo | Where-Object { $_.ModelID -like '*sd3*' }) {
                 $ModelID = $_.ModelID
                 $invokeStabilityAIImageModelSplat = @{
                     ImagesSavePath   = $outFile

diff --git a/src/Tests/Integration/SupportedModels-Checks.Tests.ps1 b/src/Tests/Integration/SupportedModels-Checks.Tests.ps1
@@ -49,7 +49,8 @@ Describe 'Supported Models Checks' -Tag Integration {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0'
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )
         $filesWithAllModelsReferences = @(
             'Add-ModelCostEstimate.ps1'

diff --git a/src/Tests/Unit/Public/Get-ModelContext.Tests.ps1 b/src/Tests/Unit/Public/Get-ModelContext.Tests.ps1
@@ -249,6 +249,13 @@ InModuleScope 'pwshBedrock' {
                             content = 'test'
                         }
                     }
+                    [PSCustomObject]@{
+                        ModelId = 'stability.sd3-5-large-v1:0'
+                        Context = [PSCustomObject]@{
+                            role    = 'user'
+                            content = 'test'
+                        }
+                    }
                 )
             }
 

diff --git a/src/Tests/Unit/Public/Get-ModelTally.Tests.ps1 b/src/Tests/Unit/Public/Get-ModelTally.Tests.ps1
@@ -43,7 +43,7 @@ InModuleScope 'pwshBedrock' {
 
             It 'should get the tally for all models' {
                 $eval = Get-ModelTally -AllModels
-                $eval.Count | Should -BeExactly 44
+                $eval.Count | Should -BeExactly 45
                 foreach ($model in $eval) {
                     if ($null -ne $model.ImageCount) {
                         $model.ImageCount | Should -BeExactly 0

diff --git a/src/Tests/Unit/Public/Reset-ModelContext.Tests.ps1 b/src/Tests/Unit/Public/Reset-ModelContext.Tests.ps1
@@ -627,6 +627,20 @@ User: "Hi there!
                             }
                         )
                     }
+                    [PSCustomObject]@{
+                        ModelID = 'stability.sd3-5-large-v1:0'
+                        Context = @(
+                            [PSCustomObject]@{
+                                role    = 'user'
+                                content = @(
+                                    [PSCustomObject]@{
+                                        type = 'text'
+                                        text = 'SD3 large v1 context'
+                                    }
+                                )
+                            }
+                        )
+                    }
                 )
             } #beforeEach
 

diff --git a/src/pwshBedrock/Imports.ps1 b/src/pwshBedrock/Imports.ps1
@@ -337,6 +337,11 @@ $Global:pwshBedRockSessionModelTally = @(
         ImageCount = 0
         ImageCost  = 0
     }
+    [PSCustomObject]@{
+        ModelId    = 'stability.sd3-5-large-v1:0'
+        ImageCount = 0
+        ImageCost  = 0
+    }
 )
 
 #endregion
@@ -520,6 +525,10 @@ $Global:pwshBedrockModelContext = @(
         ModelId = 'stability.sd3-large-v1:0'
         Context = New-Object System.Collections.Generic.List[object]
     }
+    [PSCustomObject]@{
+        ModelId = 'stability.sd3-5-large-v1:0'
+        Context = New-Object System.Collections.Generic.List[object]
+    }
 )
 
 #endregion
@@ -1691,6 +1700,33 @@ $script:stabilityAIModelInfo = @(
         # OutputTokenCost            = 0.012
         # pricing structure is different for image models
     }
+    [PSCustomObject]@{
+        ProviderName               = 'Stability AI'
+        ModelName                  = 'Stable Diffusion 3.5 Large'
+        ModelId                    = 'stability.sd3-5-large-v1:0'
+        Model                      = ''
+        Description                = 'the most powerful model in the Stable Diffusion family at 8.1 billion parameters, with superior quality and prompt adherence'
+        Strength                   = 'Story boarding, concept art creation, and rapid prototyping of visual effects. typography, intricate compositions, dynamic lighting, vibrant colors, and artistic cohesion.'
+        Multilingual               = $false
+        Text                       = $false
+        Image                      = $true
+        Video                      = $false
+        Document                   = $false
+        Vision                     = $false
+        SystemPrompt               = $false
+        ToolUse                    = $false
+        ResponseStreamingSupported = $false
+        ChatHistorySupported       = $false
+        InferenceProfile           = $false
+        ContextWindow              = ''
+        MaxOutput                  = ''
+        TrainingCutoff             = ''
+        PayloadLimit               = '' #! Couldn't find in documentation
+        ImageCost                  = 0.08
+        # InputTokenCost             = 0.01
+        # OutputTokenCost            = 0.012
+        # pricing structure is different for image models
+    }
 ) #ai21ModelInfo
 
 #endregion

diff --git a/src/pwshBedrock/Private/Add-ModelCostEstimate.ps1 b/src/pwshBedrock/Private/Add-ModelCostEstimate.ps1
@@ -97,7 +97,8 @@ function Add-ModelCostEstimate {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/Public/Get-ModelContext.ps1 b/src/pwshBedrock/Public/Get-ModelContext.ps1
@@ -78,6 +78,7 @@ function Get-ModelContext {
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
             'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0',
             'Converse'
         )]
         [string]$ModelID

diff --git a/src/pwshBedrock/Public/Get-ModelCostEstimate.ps1 b/src/pwshBedrock/Public/Get-ModelCostEstimate.ps1
@@ -109,7 +109,8 @@ function Get-ModelCostEstimate {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID
     )

diff --git a/src/pwshBedrock/Public/Get-ModelInfo.ps1 b/src/pwshBedrock/Public/Get-ModelInfo.ps1
@@ -86,7 +86,8 @@ function Get-ModelInfo {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/Public/Get-ModelTally.ps1 b/src/pwshBedrock/Public/Get-ModelTally.ps1
@@ -89,7 +89,8 @@ function Get-ModelTally {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/Public/Invoke-StabilityAIImageModel.ps1 b/src/pwshBedrock/Public/Invoke-StabilityAIImageModel.ps1
@@ -184,7 +184,8 @@ function Invoke-StabilityAIImageModel {
         [ValidateSet(
             'stability.stable-image-core-v1:0',
             'stability.stable-image-ultra-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID = 'stability.stable-image-core-v1:0',
 
@@ -244,11 +245,11 @@ function Invoke-StabilityAIImageModel {
 
     #region image-to-image parameters
 
-    if ($InitImagePath -and $ModelID -ne 'stability.sd3-large-v1:0') {
-        Write-Warning -Message 'Only stability.sd3-large-v1:0 supports image-to-image requests.'
+    if ($InitImagePath -and $ModelID -notlike '*sd3*') {
+        Write-Warning -Message 'Only stability sd3 models support image-to-image requests.'
         throw ('Model {0} does not support image-to-image requests.' -f $ModelID)
     }
-    elseif ($InitImagePath -and $ModelID -eq 'stability.sd3-large-v1:0') {
+    elseif ($InitImagePath -and $ModelID -like '*sd3*') {
         $bodyObj.Add('mode', 'image-to-image')
 
         Write-Debug -Message 'Validating InitImage'

diff --git a/src/pwshBedrock/Public/Reset-ModelContext.ps1 b/src/pwshBedrock/Public/Reset-ModelContext.ps1
@@ -83,7 +83,8 @@ function Reset-ModelContext {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/Public/Reset-ModelTally.ps1 b/src/pwshBedrock/Public/Reset-ModelTally.ps1
@@ -87,7 +87,8 @@ function Reset-ModelTally {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/Public/Save-ModelContext.ps1 b/src/pwshBedrock/Public/Save-ModelContext.ps1
@@ -75,7 +75,8 @@ function Save-ModelContext {
             'stability.stable-diffusion-xl-v1',
             'stability.stable-image-ultra-v1:0',
             'stability.stable-image-core-v1:0',
-            'stability.sd3-large-v1:0'
+            'stability.sd3-large-v1:0',
+            'stability.sd3-5-large-v1:0'
         )]
         [string]$ModelID,
 

diff --git a/src/pwshBedrock/pwshBedrock.psd1 b/src/pwshBedrock/pwshBedrock.psd1
@@ -12,7 +12,7 @@
     RootModule        = 'pwshBedrock.psm1'
 
     # Version number of this module.
-    ModuleVersion     = '0.50.0'
+    ModuleVersion     = '0.51.0'
 
     # Supported PSEditions
     # CompatiblePSEditions = @()