Merge pull request #1158 from cheald/fix-find-noise

Fix invalid invocation of find_noise_for_image
This commit is contained in:
ZeroCool 2022-09-16 04:15:31 -07:00 committed by GitHub
commit 2eaccb7b24
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -1367,7 +1367,7 @@ def process_images(
if noise_mode == 1 or noise_mode == 3:
# TODO params for find_noise_to_image
x = torch.cat(batch_size * [find_noise_for_image.find_noise_for_image(
x = torch.cat(batch_size * [find_noise_for_image(
st.session_state["model"], st.session_state["device"],
init_img.convert('RGB'), '', find_noise_steps, 0.0, normalize=True,
generation_callback=generation_callback,