diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index ad22e65c5..3db7432eb 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -727,6 +727,17 @@ model = pe.VisionTransformer.from_config("${model.id}", pretrained=True)`; return [vision_encoder]; } }; +export const phantom_wan = (model: ModelData): string[] => [ + `from huggingface_hub import snapshot_download +from phantom_wan import WANI2V, configs + +checkpoint_dir = snapshot_download("${model.id}") +wan_i2v = WanI2V( + config=configs.WAN_CONFIGS['i2v-14B'], + checkpoint_dir=checkpoint_dir, + ) + video = wan_i2v.generate(text_prompt, image_prompt)`, +]; export const pyannote_audio_pipeline = (model: ModelData): string[] => [ `from pyannote.audio import Pipeline diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index a5a043995..2e9c4957f 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -662,6 +662,14 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { snippets: snippets.perception_encoder, countDownloads: `path_extension:"pt"`, }, + "phantom-wan": { + prettyLabel: "Phantom", + repoName: "Phantom", + repoUrl: "https://github.com/Phantom-video/Phantom", + snippets: snippets.phantom_wan, + filter: false, + countDownloads: `path_extension:"pth"`, + }, pxia: { prettyLabel: "pxia", repoName: "pxia",