Update packages/tasks/src/model-libraries-snippets.ts

Co-authored-by: Lucain <[email protected]>
huggingface · Sep 3, 2024 · 3eb83db · 3eb83db
1 parent ca6264f
commit 3eb83db
Showing 1 changed file with 1 addition and 2 deletions.
diff --git 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[] => [