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