Skip to content

Commit

Permalink
codestyle fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
likholat authored and ilya-lavrenov committed Oct 9, 2024
1 parent 61e4562 commit 28d5f95
Show file tree
Hide file tree
Showing 3 changed files with 5 additions and 10 deletions.
11 changes: 5 additions & 6 deletions include/openvino/genai/text2image/pipeline.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -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>& scheduler_type, // scheduler
const CLIPTextModel& clip_text_model, // text_encoder
const UNet2DConditionModel& unet, // unet
const AutoencoderKL& vae_decoder); // vae

const std::shared_ptr<Scheduler>& 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(
Expand All @@ -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>& scheduler_type,
const CLIPTextModel& clip_text_model,
Expand Down
1 change: 0 additions & 1 deletion src/text2image/numpy_utils.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -10,7 +10,6 @@
#include <algorithm>
#include <cmath>


namespace ov {
namespace genai {
namespace numpy_utils {
Expand Down
3 changes: 0 additions & 3 deletions src/text2image/stable_diffusion_xl_pipeline.hpp
Original file line number Diff line number Diff line change
Expand Up @@ -8,7 +8,6 @@

#include "utils.hpp"


namespace ov {
namespace genai {

Expand Down Expand Up @@ -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
Expand Down

0 comments on commit 28d5f95

Please sign in to comment.