From 531524a7fd14b6087fe108e513576e8e9906bf86 Mon Sep 17 00:00:00 2001
From: BFlat <bombayapirat@gmail.com>
Date: Thu, 18 Jan 2024 20:00:11 +0700
Subject: [PATCH] sd request update

---
 .../components/src/api/_common/enums/TypeOfInference.ts    | 2 +-
 .../pages/weight/inference_panels/SdInferencePanel.tsx     | 7 ++++---
 2 files changed, 5 insertions(+), 4 deletions(-)

diff --git a/packages/components/src/api/_common/enums/TypeOfInference.ts b/packages/components/src/api/_common/enums/TypeOfInference.ts
index c942ee9dce..164a2728e2 100644
--- a/packages/components/src/api/_common/enums/TypeOfInference.ts
+++ b/packages/components/src/api/_common/enums/TypeOfInference.ts
@@ -1,5 +1,5 @@
 export enum TypeOfInference {
   UploadLora = "upload_lora",
   Checkpoint = "checkpoint",
-  Standard = "inference",
+  Inference = "inference",
 }
diff --git a/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx b/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
index 7e09afb836..3a49b4fa22 100644
--- a/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
+++ b/packages/fakeyou.com/src/v2/view/pages/weight/inference_panels/SdInferencePanel.tsx
@@ -40,7 +40,7 @@ export default function SdInferencePanel({
   sd_model_token,
 }: SdInferencePanelProps) {
   const [isEnqueuing, setIsEnqueuing] = useState(false);
-
+  const [inferenceType, inferenceTypeSet] = useState(TypeOfInference.Inference);
   const [seed, seedSet] = useState("random");
   const [seedNumber, seedNumberSet] = useState("");
   const [sampler, samplerSet] = useState("DPM++ 2M Karras");
@@ -60,6 +60,7 @@ export default function SdInferencePanel({
     setPrompt,
     setNegativePrompt,
     samplesSet,
+    inferenceTypeSet,
   });
   const [isLoraModalOpen, isLoraModalOpenSet] = useState(false);
 
@@ -208,7 +209,7 @@ export default function SdInferencePanel({
 
     const request = {
       uuid_idempotency_token: uuidv4(),
-      type_of_inference: TypeOfInference.Standard,
+      type_of_inference: inferenceType,
       maybe_sd_model_token: sd_model_token,
       maybe_lora_model_token: loraToken,
       maybe_prompt: prompt,
@@ -229,7 +230,7 @@ export default function SdInferencePanel({
     if (EnqueueImageGenIsSuccess(response)) {
       console.log("enqueuing...");
 
-      if (response.inference_job_token === "generic") {
+      if (response.inference_job_token) {
         enqueueInferenceJob(
           response.inference_job_token,
           FrontendInferenceJobType.ImageGeneration