diffusers
[Feat] add tiny Autoencoder for (almost) instant decoding
#4384
Merged
Overview
Commits
45
Changes
View On GitHub

[Feat] add tiny Autoencoder for (almost) instant decoding #4384

sayakpaul merged 45 commits into main from feat/tiny-autoenc
sayakpaul
sayakpaul add: model implementation of tiny autoencoder.
c15d27a7
sayakpaul add: inits.
937eb19d
sayakpaul push the latest devs.
19f834e9
sayakpaul Merge branch 'main' into feat/tiny-autoenc
58b9bc9d
sayakpaul add: conversion script and finish.
cddf45de
sayakpaul add: scaling factor args.
f111387a
sayakpaul debugging
258c0743
sayakpaul fix denormalization.
8be9d3d4
sayakpaul fix: positional argument.
63a95a5d
sayakpaul handle use_torch_2_0_or_xformers.
b4708976
sayakpaul handle post_quant_conv
2ce4bf32
sayakpaul handle dtype
c8ff8e5c
sayakpaul fix: sdxl image processor for tiny ae.
0c099ca4
sayakpaul fix: sdxl image processor for tiny ae.
8d43bc87
sayakpaul unify upcasting logic.
7d2dca01
sayakpaul sayakpaul requested a review from yiyixuxu yiyixuxu 2 years ago
sayakpaul sayakpaul requested a review from patrickvonplaten patrickvonplaten 2 years ago
HuggingFaceDocBuilderDev
sayakpaul copied from madness.
a7a8f6fc
sayakpaul remove trailing whitespace.
d11fd65e
madebyollin
madebyollin commented on 2023-07-31
sayakpaul set is_tiny_vae = False
92c72e24
sayakpaul address PR comments.
d6808d18
yiyixuxu
yiyixuxu approved these changes on 2023-08-01
sayakpaul change to AutoencoderTiny
5b7635b5
sayakpaul make act_fn an str throughout
4bd124d2
sayakpaul fix: apply_forward_hook decorator call
4ddc077d
sayakpaul get rid of the special is_tiny_vae flag.
ee35ebdd
sayakpaul directly scale the output.
4a606efb
sayakpaul fix dummies?
56eade87
sayakpaul
sayakpaul fix: act_fn.
ae3b1079
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
patrickvonplaten
patrickvonplaten commented on 2023-08-01
sayakpaul get rid of the Clamp() layer.
40966b80
sayakpaul bring back copied from.
0b041335
sayakpaul movement of the blocks to appropriate modules.
13f06fac
sayakpaul add: docstrings to AutoencoderTiny
ef8772a6
sayakpaul add: documentation.
8c0852c1
sayakpaul changes to the conversion script.
6f17fd38
sayakpaul add doc entry.
29e33702
sayakpaul settle tests.
817bb2b5
sayakpaul Merge branch 'main' into feat/tiny-autoenc
2995de1f
sayakpaul style
bbf0597b
sayakpaul
patrickvonplaten
patrickvonplaten commented on 2023-08-02
patrickvonplaten
patrickvonplaten approved these changes on 2023-08-02
sayakpaul add one slow test.
9105fb55
sayakpaul fix
ef3eae29
sayakpaul fix 2
e05b7303
sayakpaul fix 2
0980796a
sayakpaul fix: 4
163c0354
sayakpaul fix: 5
dd0b6736
sayakpaul finish integration tests
d7ab16fa
sayakpaul
patrickvonplaten
patrickvonplaten approved these changes on 2023-08-02
stevhliu
stevhliu approved these changes on 2023-08-02
sayakpaul Apply suggestions from code review
644d125b
sayakpaul
sayakpaul style
e2fcccb9
sayakpaul sayakpaul merged 18fc40c1 into main 2 years ago
sayakpaul sayakpaul deleted the feat/tiny-autoenc branch 2 years ago

Login to write a write a comment.

Login via GitHub

Reviewers
Assignees
No one assigned
Labels
None yet
Milestone
No milestone
Terms Of Use Privacy Policy Refund Policy Impressum