Skip to content

GitLab

"...git@developer.sourcefind.cn:cnjsdfcy/simbricks.git" did not exist on "e28dfebda14a75dedfae6d9f0a129418c5cc16a9"
Commit 0b7daa6d authored by patil-suraj's avatar patil-suraj
Browse files

add forward for vq model

parent 99568c5a
Show whitespace changes
Inline Side-by-side
Showing with 5 additions and 0 deletions
src/diffusers/models/vae.py
View file @ 0b7daa6d
...@@ -535,6 +535,11 @@ class VQModel(ModelMixin, ConfigMixin): ...@@ -535,6 +535,11 @@ class VQModel(ModelMixin, ConfigMixin):
dec = self.decoder(quant) dec = self.decoder(quant)
return dec return dec
def forward(self, x):
h = self.encode(x)
dec = self.decode(h)
return dec
class AutoencoderKL(ModelMixin, ConfigMixin): class AutoencoderKL(ModelMixin, ConfigMixin):
def __init__( def __init__(
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment