From 3eb83dbb89126296d471501a36668e528cf6edf6 Mon Sep 17 00:00:00 2001 From: NielsRogge <48327001+NielsRogge@users.noreply.github.com> Date: Tue, 3 Sep 2024 09:38:51 +0200 Subject: [PATCH] Update packages/tasks/src/model-libraries-snippets.ts Co-authored-by: Lucain --- packages/tasks/src/model-libraries-snippets.ts | 3 +-- 1 file changed, 1 insertion(+), 2 deletions(-) diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index f8125f50f..1a0a6be84 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -874,8 +874,7 @@ IWorker engine = WorkerFactory.CreateWorker(BackendType.GPUCompute, model); export const vfimamba = (model: ModelData): string[] => [ `from Trainer_finetune import Model -model = Model(-1) -model.from_pretrained("${model.id}")`, +model = Model.from_pretrained("${model.id}")`, ]; export const voicecraft = (model: ModelData): string[] => [