Revert "[Small model] Explicitly close AsyncGenerator" (huggingface#709)

ice91 · Jan 19, 2024 · a1c6c3b · a1c6c3b
1 parent 7967730
commit a1c6c3b
Showing 1 changed file with 0 additions and 1 deletion.
diff --git a/src/lib/server/generateFromDefaultEndpoint.ts b/src/lib/server/generateFromDefaultEndpoint.ts
@@ -21,7 +21,6 @@ export async function generateFromDefaultEndpoint({
 					generated_text = generated_text.slice(0, -stop.length).trimEnd();
 				}
 			}
-			tokenStream.return();
 			return generated_text;
 		}
 	}