mincomk/risuai
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

[feat] add transformer types

Browse Source
This commit is contained in:
kwaroran
2023-08-05 23:13:32 +09:00
parent aae5085f9f
commit dbbc8d25ac
1 changed files with 1 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
src/ts/process/embedding/transformers.ts
View File

@@ -1,6 +1,4 @@
import type { Pipeline, PretrainedOptions } from '@xenova/transformers'; import type { Pipeline, PretrainedOptions } from '@xenova/transformers';
import { DataBase } from 'src/ts/storage/database';
import { get } from 'svelte/store';
let pipeline: (task: string, model?: string, { quantized, progress_callback, config, cache_dir, local_files_only, revision, }?: PretrainedOptions) => Promise<Pipeline> = null let pipeline: (task: string, model?: string, { quantized, progress_callback, config, cache_dir, local_files_only, revision, }?: PretrainedOptions) => Promise<Pipeline> = null
@@ -37,7 +35,7 @@ export const runTransformers = async (baseText:string, model:string,bodyTemplate
await loadTransformer() await loadTransformer()
let text = baseText let text = baseText
let generator = await pipeline('text-generation', model); let generator = await pipeline('text-generation', model);
let output = await generator(text, bodyTemplate); let output:{generated_text:string}[] = await generator(text);
return output return output
} }