artem/stable-diffusion
1
0
Fork 0
mirror of https://github.com/CompVis/stable-diffusion.git synced 2026-04-12 03:43:34 +00:00
Code Issues Packages Projects Releases Wiki Activity

fix to numpy

Browse Source
This commit is contained in:
Patrick von Platen 2022-08-19 17:14:29 +00:00
parent 239ed0fd02
commit b9851783e5
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
scripts/txt2img.py
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")