Commit fc423594 authored by Anuraag Jain's avatar Anuraag Jain
Browse files

fix: sanitized model name

parent 5c18d61d
......@@ -314,8 +314,9 @@
};
const pullModelHandler = async () => {
if (modelDownloadStatus[modelTag]) {
toast.error(`Model '${modelTag}' is already in queue for downloading.`);
const sanitizedModelTag = modelTag.trim();
if (modelDownloadStatus[sanitizedModelTag]) {
toast.error(`Model '${sanitizedModelTag}' is already in queue for downloading.`);
return;
}
if (Object.keys(modelDownloadStatus).length === 3) {
......@@ -326,7 +327,7 @@
modelTransferring = true;
modelDownloadQueue.push(
{ modelName: modelTag },
{ modelName: sanitizedModelTag },
async (data: { modelName: string; success: boolean; error?: Error }) => {
const { modelName } = data;
// Remove the downloaded model
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment