Fix invalid invocation of find_noise_for_image

This commit is contained in:
Chris Heald 2022-09-15 12:48:35 -07:00
parent b4de6caf69
commit 51a973929c

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,