@omnimedia/omnitool
Version:
open source video processing tools
13 lines • 438 B
JavaScript
import { pipeline } from "@huggingface/transformers";
export async function loadPipe(options) {
const { spec, onLoading } = options;
const pipe = await pipeline(options.task, spec.model, {
device: spec.device,
dtype: spec.dtype,
progress_callback: (data) => {
onLoading({ total: data.total, progress: data.progress });
},
});
return pipe;
}
//# sourceMappingURL=load-pipe.js.map