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[] => [
diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts
index 96fd6329a..cac3a51dd 100644
--- a/packages/tasks/src/model-libraries.ts
+++ b/packages/tasks/src/model-libraries.ts
@@ -645,7 +645,7 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
 	},
 	"vfi-mamba": {
 		prettyLabel: "VFIMamba",
-		repoName: "vfi-mamba",
+		repoName: "VFIMamba",
 		repoUrl: "https://github.com/MCG-NJU/VFIMamba",
 		countDownloads: `path_extension:"pkl"`,
 		snippets: snippets.vfimamba,