non-square aspect ratio results in failure (although have installed all the suitable version) about hidiffusion

[ShenZhang-Shin]

I see, man. Could you let me know the resolution and model you are using?

[solitaryTian]

I see, man. Could you let me know the resolution and model you are using?

pretrain_model = "/dfs/comicai/zhiyuan.shi/models/stable-diffusion-v1-5"
h, w = 960, 512

the whole code is as follows:

from hidiffusion import apply_hidiffusion, remove_hidiffusion
from diffusers import DiffusionPipeline, DDIMScheduler, EulerAncestralDiscreteScheduler
import torch

pretrain_model = "/dfs/comicai/bin.hu/hub/darkSushiMixMix_225D"

pretrain_model = "/dfs/comicai/zhiyuan.shi/models/stable-diffusion-v1-5"
scheduler = EulerAncestralDiscreteScheduler.from_pretrained(pretrain_model, subfolder="scheduler")
pipe = DiffusionPipeline.from_pretrained(pretrain_model, scheduler = scheduler, torch_dtype=torch.float16, load_safety_checker=False).to("cuda")
pipe.safety_checker = None

# Optional. enable_xformers_memory_efficient_attention can save memory usage and increase inference speed. enable_model_cpu_offload and enable_vae_tiling can save memory usage.

pipe.enable_xformers_memory_efficient_attention()

pipe.enable_model_cpu_offload()

pipe.enable_vae_tiling()

Apply hidiffusion with a single line of code.

apply_hidiffusion(pipe)

prompt_list = [prompt1,prompt2,prompt3,prompt4,prompt5,prompt6,prompt7,prompt8,prompt9,prompt10,prompt11,prompt12,prompt13,prompt14,prompt15,prompt16,prompt17,prompt18]

seed=100
generator = torch.Generator(device=pipe.device).manual_seed(seed)
h, w = 960, 512
steps = 30
for i, prompt in enumerate(prompt_list):
image = pipe(prompt, guidance_scale=7.5, height=h, width=w, eta=1.0, generator=generator, num_inference_steps = steps).images[0]
image.save(f'pictures/h{h}w{w}{i}_seed{seed}_step{steps}.png')

[ShenZhang-Shin]

Ok, I will be going to reproduce the problem and fix it.

[solitaryTian]

Ok, I will be going to reproduce the problem and fix it.

I am very interested in this work. Have you reproduced this bug? Do you know the reason for the error and if there is any way to solve it?

[ShenZhang-Shin]

Problem solved. install the latest version
pip3 install hidiffusion==0.1.8
here is a 960x512 example with SD 1.5

[solitaryTian]

pip3 install hidiffusion==0.1.8

I have upgraded by 'pip install hidiffusion==0.1.8'. But I still get the same error

[ShenZhang-Shin]

I retest hidffusion==0.1.8 and it works fine for 960x512 resolution.
you might check your hidiffusion version,
pip3 show hidiffusion

and this my inference process

if hidiffusion==0.1.8 still gets the error. You can further check diffusers==0.27.0 or ==0.25.0