diff --git a/packages/tasks/src/model-libraries-snippets.ts b/packages/tasks/src/model-libraries-snippets.ts index 3830cdfd0..141c1d2f4 100644 --- a/packages/tasks/src/model-libraries-snippets.ts +++ b/packages/tasks/src/model-libraries-snippets.ts @@ -1035,6 +1035,13 @@ export const relik = (model: ModelData): string[] => [ relik = Relik.from_pretrained("${model.id}")`, ]; +export const renderformer = (model: ModelData): string[] => [ + `# Install from https://github.com/microsoft/renderformer + +from renderformer import RenderFormerRenderingPipeline +pipeline = RenderFormerRenderingPipeline.from_pretrained("${model.id}")`, +]; + const tensorflowttsTextToMel = (model: ModelData): string[] => [ `from tensorflow_tts.inference import AutoProcessor, TFAutoModel diff --git a/packages/tasks/src/model-libraries.ts b/packages/tasks/src/model-libraries.ts index 9641ed689..05f66b1a8 100644 --- a/packages/tasks/src/model-libraries.ts +++ b/packages/tasks/src/model-libraries.ts @@ -784,6 +784,13 @@ export const MODEL_LIBRARIES_UI_ELEMENTS = { filter: false, countDownloads: `path:"model.safetensors"`, }, + renderformer: { + prettyLabel: "RenderFormer", + repoName: "RenderFormer", + repoUrl: "https://github.com/microsoft/renderformer", + snippets: snippets.renderformer, + filter: false, + }, reverb: { prettyLabel: "Reverb", repoName: "Reverb",