[WIP] Refactor UniDiffuser Pipeline and Tests #4948
patrickvonplaten
merged
27 commits into
huggingface:main
from
dg845:unidiffuser-refactor-pipeline
Add VAE slicing and tiling methods.
a167b3fd
Switch to using VaeImageProcessing for preprocessing and postprocessi…
54ef00df
Rename the VaeImageProcessor to vae_image_processor to avoid a name c…
04c14dea
Remove the postprocess() function because we're using a VaeImageProce…
d8a1b765
Remove UniDiffuserPipeline.decode_image_latents because we're using V…
2fcac678
Refactor generating text from text latents into a decode_text_latents…
d44b0818
Add enable_full_determinism() to UniDiffuser tests.
ea348e4b
make style
2f0e6d90
sayakpaul
commented
on 2023-09-08
sayakpaul
commented
on 2023-09-08
Add PipelineLatentTesterMixin to UniDiffuserPipelineFastTests.
2f5917a1
Merge branch 'main' into unidiffuser-refactor-pipeline
5cc26ae9
Remove enable_model_cpu_offload since it is now part of DiffusionPipe…
553e93ed
Rename the VaeImageProcessor instance to self.image_processor for con…
22fb0412
Update UniDiffuser conversion script.
0b2736b8
Make safe_serialization configurable in UniDiffuser conversion script.
f23b4690
Rename image_processor to clip_image_processor in UniDiffuser tests.
4110281f
Add PipelineKarrasSchedulerTesterMixin to UniDiffuserPipelineFastTests.
ade88e25
Add initial test for compiling the UniDiffuser model (not tested yet).
1948804f
Update encode_prompt and _encode_prompt to match that of StableDiffus…
4f4d38db
Turn off standard classifier-free guidance for now.
cb37fa8f
Merge branch 'main' into unidiffuser-refactor-pipeline
24067611
Merge branch 'main' into unidiffuser-refactor-pipeline
1984f312
make style
823f465f
Merge branch 'main' into unidiffuser-refactor-pipeline
6443f8be
make fix-copies
255435c7
Merge branch 'main' into unidiffuser-refactor-pipeline
c860125e
sayakpaul
commented
on 2023-09-19
sayakpaul
commented
on 2023-09-19
sayakpaul
commented
on 2023-09-19
sayakpaul
commented
on 2023-09-19
sayakpaul
commented
on 2023-09-19
sayakpaul
approved these changes
on 2023-09-19
apply suggestions from review
8fb70b00
Merge branch 'main' into unidiffuser-refactor-pipeline
4bd7e07a
Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Login to write a write a comment.
Login via GitHub