mirror of
https://github.com/huggingface/diffusers.git
synced 2026-01-29 07:22:12 +03:00
51 lines
1.4 KiB
Python
51 lines
1.4 KiB
Python
import gc
|
|
|
|
import torch
|
|
|
|
from diffusers import StableDiffusionXLInstructPix2PixPipeline
|
|
|
|
from ..testing_utils import (
|
|
backend_empty_cache,
|
|
enable_full_determinism,
|
|
require_torch_accelerator,
|
|
slow,
|
|
torch_device,
|
|
)
|
|
|
|
|
|
enable_full_determinism()
|
|
|
|
|
|
@slow
|
|
@require_torch_accelerator
|
|
class StableDiffusionXLInstructPix2PixPipeline:
|
|
pipeline_class = StableDiffusionXLInstructPix2PixPipeline
|
|
ckpt_path = "https://huggingface.co/stabilityai/cosxl/blob/main/cosxl_edit.safetensors"
|
|
original_config = None
|
|
repo_id = "diffusers/sdxl-instructpix2pix-768"
|
|
|
|
def setup_method(self):
|
|
gc.collect()
|
|
backend_empty_cache(torch_device)
|
|
|
|
def teardown_method(self):
|
|
gc.collect()
|
|
backend_empty_cache(torch_device)
|
|
|
|
def get_inputs(self, device, generator_device="cpu", dtype=torch.float32, seed=0):
|
|
generator = torch.Generator(device=generator_device).manual_seed(seed)
|
|
inputs = {
|
|
"prompt": "a fantasy landscape, concept art, high resolution",
|
|
"generator": generator,
|
|
"num_inference_steps": 2,
|
|
"strength": 0.75,
|
|
"guidance_scale": 7.5,
|
|
"output_type": "np",
|
|
}
|
|
return inputs
|
|
|
|
def test_single_file_setting_cosxl_edit(self):
|
|
# Default is PNDM for this checkpoint
|
|
pipe = self.pipeline_class.from_single_file(self.ckpt_path, config=self.repo_id, is_cosxl_edit=True)
|
|
assert pipe.is_cosxl_edit is True
|