Skip to content

Commit 98235b6

Browse files
committed
Merge branch 'pr_6' of github.com:iefode/openvino.genai into pr_6
2 parents e23267c + 9179953 commit 98235b6

File tree

1 file changed

+6
-10
lines changed

1 file changed

+6
-10
lines changed

src/cpp/src/image_generation/flux_pipeline.hpp

+6-10
Original file line numberDiff line numberDiff line change
@@ -247,20 +247,16 @@ class FluxPipeline : public DiffusionPipeline {
247247
m_vae->reshape(num_images_per_prompt, height, width);
248248
}
249249

250-
void compile(const std::string& device, const ov::AnyMap& properties) override {
251-
update_adapters_from_properties(properties, m_generation_config.adapters);
252-
auto updated_properties = update_adapters_in_properties(properties, &FluxPipeline::derived_adapters);
253-
m_clip_text_encoder->compile(device, *updated_properties);
254-
m_t5_text_encoder->compile(device, *updated_properties);
255-
m_vae->compile(device, *updated_properties);
256-
m_transformer->compile(device, *updated_properties);
257-
}
258-
259250
void compile(const std::string& text_encode_device,
260251
const std::string& denoise_device,
261252
const std::string& vae_device,
262253
const ov::AnyMap& properties) override {
263-
OPENVINO_THROW("not supported yet.");
254+
update_adapters_from_properties(properties, m_generation_config.adapters);
255+
auto updated_properties = update_adapters_in_properties(properties, &FluxPipeline::derived_adapters);
256+
m_clip_text_encoder->compile(text_encode_device, *updated_properties);
257+
m_t5_text_encoder->compile(text_encode_device, *updated_properties);
258+
m_vae->compile(vae_device, *updated_properties);
259+
m_transformer->compile(denoise_device, *updated_properties);
264260
}
265261

266262
void compute_hidden_states(const std::string& positive_prompt, const ImageGenerationConfig& generation_config) override {

0 commit comments

Comments
 (0)