How to use diffpure extension in sd-webui version 1.7

[ZhangShao7]

Modify /extensions/sd-webui-diffpure/scripts/diffpure.py
At the first
from modules.shared import devices
to
from modules import devices

Then at the end
for pic in tqdm(pics, 'Processing pics'):
# PIL -> Tensor
img = transform((pic)).unsqueeze(0)
add
img = transform((pic)).unsqueeze(0)
img = img[:, :3, :, :] # Keep only the first three channels and discard the fourth channel

Modify /models/DiffPure/imagenet.yml
diffusion_steps and timestep_respacing to 10000
to get better pur img quality