Skip to content
Merged
Show file tree
Hide file tree
Changes from 2 commits
Commits
File filter

Filter by extension

Filter by extension

Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
7 changes: 7 additions & 0 deletions packages/tasks/src/model-libraries-snippets.ts
Original file line number Diff line number Diff line change
Expand Up @@ -871,6 +871,13 @@ 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}")`,
];

export const voicecraft = (model: ModelData): string[] => [
`from voicecraft import VoiceCraft

Expand Down
7 changes: 7 additions & 0 deletions packages/tasks/src/model-libraries.ts
Original file line number Diff line number Diff line change
Expand Up @@ -643,6 +643,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = {
filter: true,
countDownloads: `path_extension:"sentis"`,
},
vfimamba: {
prettyLabel: "VFIMamba",
repoName: "vfi-mamba",
repoUrl: "https://github.com/MCG-NJU/VFIMamba",
countDownloads: `path_extension:"pkl"`,
snippets: snippets.vfimamba,
},
voicecraft: {
prettyLabel: "VoiceCraft",
repoName: "VoiceCraft",
Expand Down