Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
renzhc
diffusers_dcu
Commits
c022e529
Unverified
Commit
c022e529
authored
Dec 22, 2023
by
Dhruv Nair
Committed by
GitHub
Dec 22, 2023
Browse files
Remove ONNX inpaint legacy (#6269)
update Co-authored-by:
Sayak Paul
<
spsayakpaul@gmail.com
>
parent
40398152
Changes
1
Show whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
0 additions
and
97 deletions
+0
-97
tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_inpaint_legacy.py
...le_diffusion/test_onnx_stable_diffusion_inpaint_legacy.py
+0
-97
No files found.
tests/pipelines/stable_diffusion/test_onnx_stable_diffusion_inpaint_legacy.py
deleted
100644 → 0
View file @
40398152
# coding=utf-8
# Copyright 2023 HuggingFace Inc.
#
# Licensed under the Apache License, Version 2.0 (the "License");
# you may not use this file except in compliance with the License.
# You may obtain a copy of the License at
#
# http://www.apache.org/licenses/LICENSE-2.0
#
# Unless required by applicable law or agreed to in writing, software
# distributed under the License is distributed on an "AS IS" BASIS,
# WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
# See the License for the specific language governing permissions and
# limitations under the License.
import
unittest
import
numpy
as
np
from
diffusers
import
OnnxStableDiffusionInpaintPipelineLegacy
from
diffusers.utils.testing_utils
import
(
is_onnx_available
,
load_image
,
load_numpy
,
nightly
,
require_onnxruntime
,
require_torch_gpu
,
)
if
is_onnx_available
():
import
onnxruntime
as
ort
@
nightly
@
require_onnxruntime
@
require_torch_gpu
class
StableDiffusionOnnxInpaintLegacyPipelineIntegrationTests
(
unittest
.
TestCase
):
@
property
def
gpu_provider
(
self
):
return
(
"CUDAExecutionProvider"
,
{
"gpu_mem_limit"
:
"15000000000"
,
# 15GB
"arena_extend_strategy"
:
"kSameAsRequested"
,
},
)
@
property
def
gpu_options
(
self
):
options
=
ort
.
SessionOptions
()
options
.
enable_mem_pattern
=
False
return
options
def
test_inference
(
self
):
init_image
=
load_image
(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
"/in_paint/overture-creations-5sI6fQgYIuo.png"
)
mask_image
=
load_image
(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
"/in_paint/overture-creations-5sI6fQgYIuo_mask.png"
)
expected_image
=
load_numpy
(
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
"/in_paint/red_cat_sitting_on_a_park_bench_onnx.npy"
)
# using the PNDM scheduler by default
pipe
=
OnnxStableDiffusionInpaintPipelineLegacy
.
from_pretrained
(
"CompVis/stable-diffusion-v1-4"
,
revision
=
"onnx"
,
safety_checker
=
None
,
feature_extractor
=
None
,
provider
=
self
.
gpu_provider
,
sess_options
=
self
.
gpu_options
,
)
pipe
.
set_progress_bar_config
(
disable
=
None
)
prompt
=
"A red cat sitting on a park bench"
generator
=
np
.
random
.
RandomState
(
0
)
output
=
pipe
(
prompt
=
prompt
,
image
=
init_image
,
mask_image
=
mask_image
,
strength
=
0.75
,
guidance_scale
=
7.5
,
num_inference_steps
=
15
,
generator
=
generator
,
output_type
=
"np"
,
)
image
=
output
.
images
[
0
]
assert
image
.
shape
==
(
512
,
512
,
3
)
assert
np
.
abs
(
expected_image
-
image
).
max
()
<
1e-2
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment