Merge branch 'add_vfimamba' of github.com:NielsRogge/huggingface.js i…

…nto add_vfimamba
huggingface · Sep 5, 2024 · 2ddf512 · 2ddf512
2 parents ed8aa94 + c319c0d
commit 2ddf512
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 3 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[] => [

diff --git 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,