@@ -247,20 +247,16 @@ class FluxPipeline : public DiffusionPipeline {
247
247
m_vae->reshape (num_images_per_prompt, height, width);
248
248
}
249
249
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
-
259
250
void compile (const std::string& text_encode_device,
260
251
const std::string& denoise_device,
261
252
const std::string& vae_device,
262
253
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);
264
260
}
265
261
266
262
void compute_hidden_states (const std::string& positive_prompt, const ImageGenerationConfig& generation_config) override {
0 commit comments