Merge pull request #324 from harmony-one/sd_images_fix_003

Sd images fix 003
harmony-one · Sep 30, 2023 · f5bcc40 · f5bcc40
2 parents ada04ab + 04ed1d8
commit f5bcc40
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/src/modules/sd-images/api/helpers.ts b/src/modules/sd-images/api/helpers.ts
@@ -193,6 +193,16 @@ export const getParamsFromPrompt = (originalPrompt: string, model?: IModel): IPa
     prompt = prompt.replace(/--model (.*)/, '')
   }
 
+  if (!modelAlias) {
+    const modelMatch = prompt.match(/--model (.*) /)
+
+    if (modelMatch) {
+      modelAlias = modelMatch[1]
+
+      prompt = prompt.replace(/--model (.*) /, '')
+    }
+  }
+
   // Add 'leogirl' to trigger /leo model
   if (model && model.name === 'leosams_helloworld' && !prompt.includes('leogirl')) {
     prompt = 'leogirl ' + prompt