Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom

Task specific OV diffusion pipelines with no mixins #889

Merged
merged 31 commits into from
Oct 9, 2024
Merged

Task specific OV diffusion pipelines with no mixins #889

Show file tree
Hide file tree
Changes from 4 commits
Commits
Show all changes
31 commits
Select commit Hold shift + click to select a range
f26c22a
added task specific pipelines
IlyasMoutawwakil Sep 8, 2024
d6932f1
added task specific tests
IlyasMoutawwakil Sep 8, 2024
a75d0db
Merge branch 'main' into ov-diffusion-pipeline-for-task
IlyasMoutawwakil Sep 8, 2024
cbba917
style
IlyasMoutawwakil Sep 8, 2024
b864894
new diffusers modeling
IlyasMoutawwakil Sep 17, 2024
793001e
Merge branch 'main' into ov-diffusion-pipeline-for-task
IlyasMoutawwakil Sep 17, 2024
627377c
style
IlyasMoutawwakil Sep 17, 2024
a3c253c
fix
IlyasMoutawwakil Sep 17, 2024
1292596
fix decoder reshape
IlyasMoutawwakil Sep 17, 2024
6c551a2
test
IlyasMoutawwakil Sep 17, 2024
e1e37b6
fix and extend tests to all possible input types
IlyasMoutawwakil Sep 17, 2024
ef13139
only translate generators
IlyasMoutawwakil Sep 17, 2024
ddf913f
dummy objects
IlyasMoutawwakil Sep 17, 2024
f577cc7
fix import
IlyasMoutawwakil Sep 17, 2024
0aff7b0
patch work
IlyasMoutawwakil Sep 19, 2024
2ab13d5
update tests
IlyasMoutawwakil Sep 19, 2024
f62d7cd
fixes for many tests
IlyasMoutawwakil Sep 19, 2024
1830312
fix
IlyasMoutawwakil Sep 19, 2024
15bcdf5
test
IlyasMoutawwakil Sep 19, 2024
2a42c9d
fix
IlyasMoutawwakil Sep 19, 2024
70c69cf
Merge branch 'main' into ov-diffusion-pipeline-for-task
IlyasMoutawwakil Oct 3, 2024
faed59c
update
IlyasMoutawwakil Oct 7, 2024
d5dbc5d
Merge branch 'ov-diffusion-pipeline-for-task' of https://github.com/h…
IlyasMoutawwakil Oct 7, 2024
69eeeda
Merge branch 'main' into ov-diffusion-pipeline-for-task
IlyasMoutawwakil Oct 8, 2024
8e7c4fd
warning for missing config attr
IlyasMoutawwakil Oct 8, 2024
1f6d7fd
added test_safety_checker and test_height_width_properties
IlyasMoutawwakil Oct 8, 2024
89813bc
fix static shapes test
IlyasMoutawwakil Oct 8, 2024
ed8f3e4
fix quantization
IlyasMoutawwakil Oct 8, 2024
6339a9e
remove the need for a base OVpipeline class
IlyasMoutawwakil Oct 9, 2024
c540c7b
fix dispatch
IlyasMoutawwakil Oct 9, 2024
354abdf
fix import
IlyasMoutawwakil Oct 9, 2024
File filter

Filter by extension

Filter by extension
Conversations
Failed to load comments.
Loading
Jump to
Jump to file
Failed to load files.
Loading
Diff view
Diff view
16 changes: 16 additions & 0 deletions optimum/intel/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -100,6 +100,10 @@
"OVStableDiffusionXLPipeline",
"OVStableDiffusionXLImg2ImgPipeline",
"OVLatentConsistencyModelPipeline",
"OVPipelineForImage2Image",
"OVPipelineForText2Image",
"OVPipelineForInpainting",
"OVDiffusionPipeline",
]
else:
_import_structure["openvino"].extend(
Expand All @@ -110,6 +114,10 @@
"OVStableDiffusionXLPipeline",
"OVStableDiffusionXLImg2ImgPipeline",
"OVLatentConsistencyModelPipeline",
"OVPipelineForImage2Image",
"OVPipelineForText2Image",
"OVPipelineForInpainting",
"OVDiffusionPipeline",
]
)

Expand Down Expand Up @@ -246,7 +254,11 @@
raise OptionalDependencyNotAvailable()
except OptionalDependencyNotAvailable:
from .utils.dummy_openvino_and_diffusers_objects import (
OVDiffusionPipeline,
OVLatentConsistencyModelPipeline,
OVPipelineForImage2Image,
OVPipelineForInpainting,
OVPipelineForText2Image,
OVStableDiffusionImg2ImgPipeline,
OVStableDiffusionInpaintPipeline,
OVStableDiffusionPipeline,
Expand All @@ -255,7 +267,11 @@
)
else:
from .openvino import (
OVDiffusionPipeline,
OVLatentConsistencyModelPipeline,
OVPipelineForImage2Image,
OVPipelineForInpainting,
OVPipelineForText2Image,
OVStableDiffusionImg2ImgPipeline,
OVStableDiffusionInpaintPipeline,
OVStableDiffusionPipeline,
Expand Down
4 changes: 4 additions & 0 deletions optimum/intel/openvino/__init__.py
View file
Open in desktop
Original file line number Diff line number Diff line change
Expand Up @@ -75,7 +75,11 @@

if is_diffusers_available():
from .modeling_diffusion import (
OVDiffusionPipeline,
OVLatentConsistencyModelPipeline,
OVPipelineForImage2Image,
OVPipelineForInpainting,
OVPipelineForText2Image,
OVStableDiffusionImg2ImgPipeline,
OVStableDiffusionInpaintPipeline,
OVStableDiffusionPipeline,
Expand Down
Loading
Loading