diff --git a/src/models/huggingface.js b/src/models/huggingface.js index 59dd0f7..80c36e8 100644 --- a/src/models/huggingface.js +++ b/src/models/huggingface.js @@ -55,7 +55,7 @@ export class HuggingFace { const hasCloseTag = res.includes(""); // If there's a partial mismatch, warn and retry the entire request. - if ((hasOpenTag && !hasCloseTag) || (!hasOpenTag && hasCloseTag)) { + if ((hasOpenTag && !hasCloseTag)) { console.warn("Partial block detected. Re-generating..."); continue; }