@tryhamster/gerbil 1.6.2 → 1.7.0

package/dist/gpu/hooks.mjs

@@ -1,4 +1,4 @@
-import { i as resolveDefaultRepo, r as OUTETTS_PRESET_VOICES, t as DEFAULT_MODELS } from "../defaults-KnrRSIeJ.mjs";
+import { i as resolveDefaultRepo, r as OUTETTS_PRESET_VOICES, t as DEFAULT_MODELS } from "../defaults-C_bJK9zs.mjs";
 import { useCallback, useEffect, useRef, useState } from "react";
 import { Fragment, jsx } from "react/jsx-runtime";
 
@@ -271,7 +271,10 @@ function useEngine(options = {}) {
 		try {
 			const result = await engine.generate(prompt, {
 				maxTokens: opts.maxTokens ?? 256,
-				sampling: { temperature: opts.temperature ?? .7 },
+				sampling: {
+					temperature: opts.temperature ?? .7,
+					repetitionPenalty: opts.repetitionPenalty
+				},
 				systemPrompt: opts.system,
 				stopSequences: opts.stopSequences,
 				onToken: (token, meta) => {
@@ -318,7 +321,10 @@ function useEngine(options = {}) {
 				schema: opts.schema,
 				maxRetries: opts.maxRetries,
 				maxTokens: opts.maxTokens ?? 256,
-				sampling: { temperature: opts.temperature ?? .7 },
+				sampling: {
+					temperature: opts.temperature ?? .7,
+					repetitionPenalty: opts.repetitionPenalty
+				},
 				systemPrompt: opts.system,
 				stopSequences: opts.stopSequences
 			});
@@ -348,7 +354,10 @@ function useEngine(options = {}) {
 		try {
 			const result = await engine.describeImage(decoded, prompt, {
 				maxTokens: opts.maxTokens ?? 150,
-				sampling: { temperature: opts.temperature ?? .7 },
+				sampling: {
+					temperature: opts.temperature ?? .7,
+					repetitionPenalty: opts.repetitionPenalty ?? 1.3
+				},
 				systemPrompt: opts.system,
 				stopSequences: opts.stopSequences,
 				onToken: (token, meta) => {
@@ -646,7 +655,7 @@ function useMemory(options = {}) {
 		if (initRef.current) return initRef.current;
 		initRef.current = (async () => {
 			if (!embedder.isReady) await embedder.load();
-			const [{ createMemory }, { createIndexedDBStore }] = await Promise.all([import("../memory-D1P7Tmda.mjs"), import("../indexeddb-store-ClH12Xnl.mjs")]);
+			const [{ createMemory }, { createIndexedDBStore }] = await Promise.all([import("../memory-BwYh_DZh.mjs"), import("../indexeddb-store-B6HYVgQ_.mjs")]);
 			const mem = createMemory({
 				embed: async (texts) => Promise.all(texts.map((t) => embedder.embed(t))),
 				store: createIndexedDBStore({ dbName: namespace })