novarobot
/
stable-diffusion-cpuonly
mirror of https://github.com/novarobot/stable-diffusion-cpuonly
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

fix to numpy

Browse Source
main
Patrick von Platen 3 years ago
parent 239ed0fd02
commit b9851783e5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File

2
scripts/txt2img.py
Unescape Escape View File

@ -266,7 +266,7 @@ def main():
x_samples_ddim = model.decode_first_stage(samples_ddim) x_samples_ddim = model.decode_first_stage(samples_ddim)
x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0) x_samples_ddim = torch.clamp((x_samples_ddim + 1.0) / 2.0, min=0.0, max=1.0)
x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1) x_samples_ddim = x_samples_ddim.cpu().permute(0, 2, 3, 1).numpy()
x_image = x_samples_ddim x_image = x_samples_ddim
safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt") safety_checker_input = safety_feature_extractor(numpy_to_pil(x_image), return_tensors="pt")

Write Preview
Loading…
Cancel
Save