MCPcopy Create free account
hub / github.com/bbaaii/DreamDiffusion / encode_to_prequant

Method encode_to_prequant

code/dc_ldm/models/autoencoder.py:223–226  ·  view source on GitHub ↗
(self, x)

Source from the content-addressed store, hash-verified

221 return quant, emb_loss, info
222
223 def encode_to_prequant(self, x):
224 h = self.encoder(x)
225 h = self.quant_conv(h)
226 return h
227
228 def decode(self, quant):
229 quant = self.post_quant_conv(quant)

Callers

nothing calls this directly

Calls

no outgoing calls

Tested by

no test coverage detected