Skip to content

Commit e889326

Browse files
authored
Add pip install command to diffusers snippets (#1805)
1 parent e80852f commit e889326

File tree

1 file changed

+16
-11
lines changed

1 file changed

+16
-11
lines changed

packages/tasks/src/model-libraries-snippets.ts

Lines changed: 16 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -324,6 +324,8 @@ dam = DescribeAnythingModel(
324324
)`,
325325
];
326326

327+
const diffusers_install = "pip install -U diffusers transformers";
328+
327329
const diffusersDefaultPrompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k";
328330

329331
const diffusersImg2ImgDefaultPrompt = "Turn this cat into a dog";
@@ -485,34 +487,37 @@ image = pipe(
485487
];
486488

487489
export const diffusers = (model: ModelData): string[] => {
490+
let codeSnippets: string[];
488491
if (
489492
model.tags.includes("StableDiffusionInpaintPipeline") ||
490493
model.tags.includes("StableDiffusionXLInpaintPipeline")
491494
) {
492-
return diffusers_inpainting(model);
495+
codeSnippets = diffusers_inpainting(model);
493496
} else if (model.tags.includes("controlnet")) {
494-
return diffusers_controlnet(model);
497+
codeSnippets = diffusers_controlnet(model);
495498
} else if (model.tags.includes("lora")) {
496499
if (model.pipeline_tag === "image-to-image") {
497-
return diffusers_lora_image_to_image(model);
500+
codeSnippets = diffusers_lora_image_to_image(model);
498501
} else if (model.pipeline_tag === "image-to-video") {
499-
return diffusers_lora_image_to_video(model);
502+
codeSnippets = diffusers_lora_image_to_video(model);
500503
} else if (model.pipeline_tag === "text-to-video") {
501-
return diffusers_lora_text_to_video(model);
504+
codeSnippets = diffusers_lora_text_to_video(model);
502505
} else {
503-
return diffusers_lora(model);
506+
codeSnippets = diffusers_lora(model);
504507
}
505508
} else if (model.tags.includes("textual_inversion")) {
506-
return diffusers_textual_inversion(model);
509+
codeSnippets = diffusers_textual_inversion(model);
507510
} else if (model.tags.includes("FluxFillPipeline")) {
508-
return diffusers_flux_fill(model);
511+
codeSnippets = diffusers_flux_fill(model);
509512
} else if (model.pipeline_tag === "image-to-video") {
510-
return diffusers_image_to_video(model);
513+
codeSnippets = diffusers_image_to_video(model);
511514
} else if (model.pipeline_tag === "image-to-image") {
512-
return diffusers_image_to_image(model);
515+
codeSnippets = diffusers_image_to_image(model);
513516
} else {
514-
return diffusers_default(model);
517+
codeSnippets = diffusers_default(model);
515518
}
519+
520+
return [diffusers_install, ...codeSnippets];
516521
};
517522

518523
export const diffusionkit = (model: ModelData): string[] => {

0 commit comments

Comments
 (0)