diff --git a/include/openvino/genai/text2image/pipeline.hpp b/include/openvino/genai/text2image/pipeline.hpp index 3a47ce3..5ce6a08 100644 --- a/include/openvino/genai/text2image/pipeline.hpp +++ b/include/openvino/genai/text2image/pipeline.hpp @@ -110,11 +110,10 @@ class OPENVINO_GENAI_EXPORTS Text2ImagePipeline { // creates either LCM or SD pipeline from building blocks static Text2ImagePipeline stable_diffusion( - const std::shared_ptr& scheduler_type, // scheduler - const CLIPTextModel& clip_text_model, // text_encoder - const UNet2DConditionModel& unet, // unet - const AutoencoderKL& vae_decoder); // vae - + const std::shared_ptr& scheduler_type, + const CLIPTextModel& clip_text_model, + const UNet2DConditionModel& unet, + const AutoencoderKL& vae_decoder); // creates either LCM or SD pipeline from building blocks static Text2ImagePipeline latent_consistency_model( @@ -123,7 +122,7 @@ class OPENVINO_GENAI_EXPORTS Text2ImagePipeline { const UNet2DConditionModel& unet, const AutoencoderKL& vae_decoder); - // creates either SDXL pipeline from building blocks + // creates SDXL pipeline from building blocks static Text2ImagePipeline stable_diffusion_xl( const std::shared_ptr& scheduler_type, const CLIPTextModel& clip_text_model, diff --git a/src/text2image/numpy_utils.hpp b/src/text2image/numpy_utils.hpp index b95d9e4..d6144ee 100644 --- a/src/text2image/numpy_utils.hpp +++ b/src/text2image/numpy_utils.hpp @@ -10,7 +10,6 @@ #include #include - namespace ov { namespace genai { namespace numpy_utils { diff --git a/src/text2image/stable_diffusion_xl_pipeline.hpp b/src/text2image/stable_diffusion_xl_pipeline.hpp index 3d42a83..95ea2ab 100644 --- a/src/text2image/stable_diffusion_xl_pipeline.hpp +++ b/src/text2image/stable_diffusion_xl_pipeline.hpp @@ -8,7 +8,6 @@ #include "utils.hpp" - namespace ov { namespace genai { @@ -198,8 +197,6 @@ class Text2ImagePipeline::StableDiffusionXLPipeline : public Text2ImagePipeline: } } - - // replicate encoder hidden state to UNet model if (generation_config.num_images_per_prompt == 1) { // reuse output of text encoder directly w/o extra memory copy