PR #1824 Flax: Fix img2img and align with other pipeline

Flax: Fix img2img and align with other pipeline #1824

pcuenca merged 11 commits into huggingface:main from skirsten:flax/fix-img2img

pcuenca commented on 2022-12-25

pcuenca requested a review from

yiyixuxu 2 years ago

pcuenca requested a review from

patil-suraj 2 years ago

patil-suraj commented on 2022-12-26

patil-suraj approved these changes on 2022-12-26

pcuenca approved these changes on 2022-12-29

Flax: Add components function

1eb9024e

Flax: Fix img2img and align with other pipeline

6bf1983c

Flax: Fix PRNGKey type

77534313

Refactor strength to start_timestep

9d109818

Fix preprocess images

a2766393

Fix processed_images dimen

1f0117d3

latents.shape -> latents_shape

da3311d2

Fix typo

44546ef5

Remove "static" comment

74128b2a

Remove unnecessary optional types in _generate

19b3de46

skirsten force pushed from 7085ba13 to 19b3de46 2 years ago

Apply doc-builder code style.

0d937ead

pcuenca merged ab0e92fd into main 2 years ago

Reviewers

pcuenca

patil-suraj

yiyixuxu

Assignees

No one assigned

Labels

None yet

Milestone

No milestone