Something went wrong. Try again.
a tiny pet with a GPT-2 brain, living in your browser
Something went wrong. Try again.
12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667/** The small surface used from the separately vendored transformers.js runtime. * This is a type boundary, not a second runtime dependency or implementation. */export interface Tensor { dims: number[];}export interface Tokenizer { (text: string): { input_ids: Tensor; attention_mask: Tensor }; encode(text: string): number[]; decode(ids: number[], options: { skip_special_tokens: boolean }): string;}export interface Progress { status: string; file: string; loaded: number; total: number;}export interface Stopper { _call(ids: (number | bigint)[][]): boolean[];}export interface Model { generate(options: { input_ids: Tensor; attention_mask: Tensor; max_new_tokens: number; do_sample: boolean; temperature?: number; top_k?: number; top_p?: number; stopping_criteria?: Stopper[]; streamer?: object; }): Promise<unknown>;}export interface Transformers { env: { allowRemoteModels: boolean; allowLocalModels: boolean; localModelPath: string; useBrowserCache: boolean; backends: { onnx: { wasm: { numThreads: number; wasmPaths: string }; logLevel: string } }; }; AutoTokenizer: { from_pretrained( id: string, options: { progress_callback: (p: Progress) => void }, ): Promise<Tokenizer>; }; AutoModelForCausalLM: { from_pretrained( id: string, options: { device: string; dtype: string; progress_callback: (p: Progress) => void; session_options: { extra: { session: { disable_quant_qdq: string; disable_prepacking: string } }; }; }, ): Promise<Model>; }; StoppingCriteria: new () => Stopper; TextStreamer: new ( tokenizer: Tokenizer, options: { skip_prompt: boolean; callback_function: () => void }, ) => object;}