diff --git a/src/renderer/lib/transformerlab-api-sdk.ts b/src/renderer/lib/transformerlab-api-sdk.ts index 893daff988e7487d1a5035b3e0f5e9aca4e77680..43c37615bdc6735b4b14c556306d24cfe58f2e65 100644 --- a/src/renderer/lib/transformerlab-api-sdk.ts +++ b/src/renderer/lib/transformerlab-api-sdk.ts @@ -742,14 +742,22 @@ export async function getAvailableModels() { return result; } -export async function downloadModelFromHuggingFace(modelName: string) { +export async function downloadModelFromHuggingFace( + modelName: string, + job_id = null +) { + console.log(encodeURIComponent(modelName)); + + let requestString = `${API_URL()}model/download_from_huggingface?model=${encodeURIComponent( + modelName + )}`; + if (job_id) { + requestString += `&job_id=${job_id}`; + } + let result = {}; try { - const response = await fetch( - `${API_URL()}model/download_from_huggingface?model=${encodeURIComponent( - modelName - )}` - ); + const response = await fetch(requestString); result = await response.json(); // Error during fetch