diff --git a/packages/components/src/api/image_generation/UploadLora.ts b/packages/components/src/api/image_generation/UploadLora.ts new file mode 100644 index 0000000000..39e14c8ed4 --- /dev/null +++ b/packages/components/src/api/image_generation/UploadLora.ts @@ -0,0 +1,22 @@ +import MakeRequest from "../MakeRequest"; + +export interface UploadLoraRequest { + uuid_idempotency_token: String, + //type_of_inference: String, // upload_lora / check_point / inference + //maybe_upload_path?: String, + maybe_lora_upload_path?: String, + maybe_name: String, + maybe_description: String, + visibility: String, +} + +export interface UploadLoraResponse { + success: boolean, + maybe_sd_model_token?: String, + maybe_lora_model_token?: String, +} + +export const UploadLora = MakeRequest({ + method: "POST", + routingFunction: () => "/v1/image_gen/upload/lora", +}); \ No newline at end of file diff --git a/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx b/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx index bcc046a5fa..52d221e7a1 100644 --- a/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx +++ b/packages/fakeyou.com/src/hooks/useLoraUpload/useLoraUpload.tsx @@ -1,5 +1,5 @@ import { useState } from "react"; -import { UploadModel } from "@storyteller/components/src/api/image_generation/UploadModel"; +import { UploadLora } from "@storyteller/components/src/api/image_generation/UploadLora"; import { FetchStatus } from "@storyteller/components/src/api/_common/SharedFetchTypes"; import { useCoverImgUpload, useInferenceJobs } from "hooks"; import { FrontendInferenceJobType } from "@storyteller/components/src/jobs/InferenceJob"; @@ -28,15 +28,15 @@ export default function useLoraUpload() { const upload = () => { writeStatusSet(FetchStatus.in_progress); - UploadModel("", { - ...(coverImg.token - ? { cover_image_media_file_token: coverImg.token } - : {}), + UploadLora("", { + //...(coverImg.token + // ? { cover_image_media_file_token: coverImg.token } + // : {}), maybe_name: title, maybe_description: descriptionMD, uuid_idempotency_token: uuidv4(), - type_of_inference: "inference", - maybe_upload_path: uploadPath, + //type_of_inference: "inference", + maybe_lora_upload_path: uploadPath, visibility, }) .then((res: any) => {