Spaces:

radames
/

Candle-Moondream-2

Running

radames commited on Apr 2

Commit

2862e32

•

1 Parent(s): 9f3b1ae

Update moondreamWorker.js

Files changed (1) hide show

moondreamWorker.js CHANGED Viewed

@@ -164,7 +164,7 @@ async function generate(data) {
         }
         const token = await model.next_token();
         console.log("Token: ", token);
-        if (token === "<|endoftext|>") {
           self.postMessage({
             status: "complete",
             message: "complete",

         }
         const token = await model.next_token();
         console.log("Token: ", token);
+        if (token === "<|endoftext|>" || token === "END") {
           self.postMessage({
             status: "complete",
             message: "complete",