diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index c33cc48843..4c979840d8 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -2112,6 +2112,12 @@ export const hezar = (model: ModelData): string[] => [ model = Model.load("${model.id}")`, ]; +export const hunyuanimage_2_1 = (model: ModelData): string[] => [ + `from hyimage.diffusion.pipelines.hunyuanimage_pipeline import HunyuanImagePipeline +pipe = HunyuanImagePipeline.from_pretrained("${model.id}")`, +]; + + export const zonos = (model: ModelData): string[] => [ `# pip install git+https://github.com/Zyphra/Zonos.git import torchaudio diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 45f9bb6db3..0834559456 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -1261,6 +1261,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { countDownloads: `path:"model_vae_fp16.pt"`, snippets: snippets.threedtopia_xl, }, + "HunyuanImage-2.1":{ + prettyLabel: "HunyuanImage-2.1", + repoName: "HunyuanImage-2.1", + repoUrl: "https://huggingface.co/tencent/HunyuanImage-2.1", + countDownloads: `path_filename:"hunyuanimage2.1.safetensors"`, + snippets: snippets.hunyuanimage_2_1, + } } satisfies Record; export type ModelLibraryKey = keyof typeof MODEL_LIBRARIES_UI_ELEMENTS;