Fanli Lin
6b288ec44d
* enable on xpu * add 1 more * add one more * enable more * add 1 more * add more * enable 1 * enable more cases * enable * enable * update comment * one more * enable 1 * add more cases * enable xpu * add one more caswe * add more cases * add 1 * add more * add more cases * add case * enable * add more * add more * add more * enbale more * add more * update code * update test marker * add skip back * update comment * remove single files * remove * style * add * revert * reformat * enable * enable esingle g * add 2 more * update decorator * update * update * update * Update tests/pipelines/deepfloyd_if/test_if.py Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com> * Update src/diffusers/utils/testing_utils.py Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com> * Update tests/pipelines/animatediff/test_animatediff_controlnet.py Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com> * Update tests/pipelines/animatediff/test_animatediff.py Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com> * Update tests/pipelines/animatediff/test_animatediff_controlnet.py Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com> * update float16 * no unitest.skipt * update * apply style check * adapt style --------- Co-authored-by: Sayak Paul <spsayakpaul@gmail.com> Co-authored-by: Dhruv Nair <dhruv.nair@gmail.com>
71 lines
2.4 KiB
Python
71 lines
2.4 KiB
Python
import gc
|
|
import unittest
|
|
|
|
import torch
|
|
|
|
from diffusers import (
|
|
StableDiffusionUpscalePipeline,
|
|
)
|
|
from diffusers.utils import load_image
|
|
from diffusers.utils.testing_utils import (
|
|
backend_empty_cache,
|
|
enable_full_determinism,
|
|
numpy_cosine_similarity_distance,
|
|
require_torch_accelerator,
|
|
slow,
|
|
torch_device,
|
|
)
|
|
|
|
from .single_file_testing_utils import SDSingleFileTesterMixin
|
|
|
|
|
|
enable_full_determinism()
|
|
|
|
|
|
@slow
|
|
@require_torch_accelerator
|
|
class StableDiffusionUpscalePipelineSingleFileSlowTests(unittest.TestCase, SDSingleFileTesterMixin):
|
|
pipeline_class = StableDiffusionUpscalePipeline
|
|
ckpt_path = "https://huggingface.co/stabilityai/stable-diffusion-x4-upscaler/blob/main/x4-upscaler-ema.safetensors"
|
|
original_config = "https://raw.githubusercontent.com/Stability-AI/stablediffusion/main/configs/stable-diffusion/x4-upscaling.yaml"
|
|
repo_id = "stabilityai/stable-diffusion-x4-upscaler"
|
|
|
|
def setUp(self):
|
|
super().setUp()
|
|
gc.collect()
|
|
backend_empty_cache(torch_device)
|
|
|
|
def tearDown(self):
|
|
super().tearDown()
|
|
gc.collect()
|
|
backend_empty_cache(torch_device)
|
|
|
|
def test_single_file_format_inference_is_same_as_pretrained(self):
|
|
image = load_image(
|
|
"https://huggingface.co/datasets/hf-internal-testing/diffusers-images/resolve/main"
|
|
"/sd2-upscale/low_res_cat.png"
|
|
)
|
|
|
|
prompt = "a cat sitting on a park bench"
|
|
pipe = StableDiffusionUpscalePipeline.from_pretrained(self.repo_id)
|
|
pipe.enable_model_cpu_offload()
|
|
|
|
generator = torch.Generator("cpu").manual_seed(0)
|
|
output = pipe(prompt=prompt, image=image, generator=generator, output_type="np", num_inference_steps=3)
|
|
image_from_pretrained = output.images[0]
|
|
|
|
pipe_from_single_file = StableDiffusionUpscalePipeline.from_single_file(self.ckpt_path)
|
|
pipe_from_single_file.enable_model_cpu_offload()
|
|
|
|
generator = torch.Generator("cpu").manual_seed(0)
|
|
output_from_single_file = pipe_from_single_file(
|
|
prompt=prompt, image=image, generator=generator, output_type="np", num_inference_steps=3
|
|
)
|
|
image_from_single_file = output_from_single_file.images[0]
|
|
|
|
assert image_from_pretrained.shape == (512, 512, 3)
|
|
assert image_from_single_file.shape == (512, 512, 3)
|
|
assert (
|
|
numpy_cosine_similarity_distance(image_from_pretrained.flatten(), image_from_single_file.flatten()) < 1e-3
|
|
)
|