openvino_genai.FluxTransformer2DModel#

class openvino_genai.FluxTransformer2DModel#

Bases: pybind11_object

FluxTransformer2DModel class.

__init__(*args, **kwargs)#

Overloaded function.

  1. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, root_dir: os.PathLike) -> None

    FluxTransformer2DModel class root_dir (os.PathLike): Model root directory.

  2. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, root_dir: os.PathLike, device: str, **kwargs) -> None

    UNet2DConditionModel class root_dir (os.PathLike): Model root directory. device (str): Device on which inference will be done. kwargs: Device properties.

  3. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, model: openvino_genai.py_openvino_genai.FluxTransformer2DModel) -> None

FluxTransformer2DModel model

FluxTransformer2DModel class model (FluxTransformer2DModel): FluxTransformer2DModel model

Methods

__delattr__(name, /)

Implement delattr(self, name).

__dir__()

Default dir() implementation.

__eq__(value, /)

Return self==value.

__format__(format_spec, /)

Default object formatter.

__ge__(value, /)

Return self>=value.

__getattribute__(name, /)

Return getattr(self, name).

__gt__(value, /)

Return self>value.

__hash__()

Return hash(self).

__init__(*args, **kwargs)

Overloaded function.

__init_subclass__

This method is called when a class is subclassed.

__le__(value, /)

Return self<=value.

__lt__(value, /)

Return self<value.

__ne__(value, /)

Return self!=value.

__new__(**kwargs)

__reduce__()

Helper for pickle.

__reduce_ex__(protocol, /)

Helper for pickle.

__repr__()

Return repr(self).

__setattr__(name, value, /)

Implement setattr(self, name, value).

__sizeof__()

Size of object in memory, in bytes.

__str__()

Return str(self).

__subclasshook__

Abstract classes can override this to customize issubclass().

compile(self, device, **kwargs)

Compiles the model.

get_config(self)

infer(self, sample, timestep)

reshape(self, batch_size, height, width, ...)

set_hidden_states(self, tensor_name, ...)

class Config#

Bases: pybind11_object

This class is used for storing FluxTransformer2DModel config.

__class__#

alias of pybind11_type

__delattr__(name, /)#

Implement delattr(self, name).

__dir__()#

Default dir() implementation.

__eq__(value, /)#

Return self==value.

__format__(format_spec, /)#

Default object formatter.

__ge__(value, /)#

Return self>=value.

__getattribute__(name, /)#

Return getattr(self, name).

__gt__(value, /)#

Return self>value.

__hash__()#

Return hash(self).

__init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel.Config, config_path: os.PathLike) None#
__init_subclass__()#

This method is called when a class is subclassed.

The default implementation does nothing. It may be overridden to extend subclasses.

__le__(value, /)#

Return self<=value.

__lt__(value, /)#

Return self<value.

__ne__(value, /)#

Return self!=value.

__new__(**kwargs)#
__reduce__()#

Helper for pickle.

__reduce_ex__(protocol, /)#

Helper for pickle.

__repr__()#

Return repr(self).

__setattr__(name, value, /)#

Implement setattr(self, name, value).

__sizeof__()#

Size of object in memory, in bytes.

__str__()#

Return str(self).

__subclasshook__()#

Abstract classes can override this to customize issubclass().

This is invoked early on by abc.ABCMeta.__subclasscheck__(). It should return True, False or NotImplemented. If it returns NotImplemented, the normal algorithm is used. Otherwise, it overrides the normal algorithm (and the outcome is cached).

property default_sample_size#
property in_channels#
__class__#

alias of pybind11_type

__delattr__(name, /)#

Implement delattr(self, name).

__dir__()#

Default dir() implementation.

__eq__(value, /)#

Return self==value.

__format__(format_spec, /)#

Default object formatter.

__ge__(value, /)#

Return self>=value.

__getattribute__(name, /)#

Return getattr(self, name).

__gt__(value, /)#

Return self>value.

__hash__()#

Return hash(self).

__init__(*args, **kwargs)#

Overloaded function.

  1. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, root_dir: os.PathLike) -> None

    FluxTransformer2DModel class root_dir (os.PathLike): Model root directory.

  2. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, root_dir: os.PathLike, device: str, **kwargs) -> None

    UNet2DConditionModel class root_dir (os.PathLike): Model root directory. device (str): Device on which inference will be done. kwargs: Device properties.

  3. __init__(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, model: openvino_genai.py_openvino_genai.FluxTransformer2DModel) -> None

FluxTransformer2DModel model

FluxTransformer2DModel class model (FluxTransformer2DModel): FluxTransformer2DModel model

__init_subclass__()#

This method is called when a class is subclassed.

The default implementation does nothing. It may be overridden to extend subclasses.

__le__(value, /)#

Return self<=value.

__lt__(value, /)#

Return self<value.

__ne__(value, /)#

Return self!=value.

__new__(**kwargs)#
__reduce__()#

Helper for pickle.

__reduce_ex__(protocol, /)#

Helper for pickle.

__repr__()#

Return repr(self).

__setattr__(name, value, /)#

Implement setattr(self, name, value).

__sizeof__()#

Size of object in memory, in bytes.

__str__()#

Return str(self).

__subclasshook__()#

Abstract classes can override this to customize issubclass().

This is invoked early on by abc.ABCMeta.__subclasscheck__(). It should return True, False or NotImplemented. If it returns NotImplemented, the normal algorithm is used. Otherwise, it overrides the normal algorithm (and the outcome is cached).

compile(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, device: str, **kwargs) None#

Compiles the model. device (str): Device to run the model on (e.g., CPU, GPU). kwargs: Device properties.

get_config(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel) openvino_genai.py_openvino_genai.FluxTransformer2DModel.Config#
infer(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, sample: openvino._pyopenvino.Tensor, timestep: openvino._pyopenvino.Tensor) openvino._pyopenvino.Tensor#
reshape(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, batch_size: int, height: int, width: int, tokenizer_model_max_length: int) openvino_genai.py_openvino_genai.FluxTransformer2DModel#
set_hidden_states(self: openvino_genai.py_openvino_genai.FluxTransformer2DModel, tensor_name: str, encoder_hidden_states: openvino._pyopenvino.Tensor) None#