@@ -324,6 +324,8 @@ dam = DescribeAnythingModel(
324324)` ,
325325] ;
326326
327+ const diffusers_install = "pip install -U diffusers transformers" ;
328+
327329const diffusersDefaultPrompt = "Astronaut in a jungle, cold color palette, muted colors, detailed, 8k" ;
328330
329331const diffusersImg2ImgDefaultPrompt = "Turn this cat into a dog" ;
@@ -485,34 +487,37 @@ image = pipe(
485487] ;
486488
487489export 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
518523export const diffusionkit = ( model : ModelData ) : string [ ] => {
0 commit comments