Function vae_encode

models/cosmos_predict2.py:102–103 · view source on GitHub ↗

(tensor, vae)

Source from the content-addressed store, hash-verified

100
101
102	def vae_encode(tensor, vae):
103	return vae.model.encode(tensor, vae.scale)
104
105
106	def get_dit_config(state_dict, key_prefix=''):

fnMethod · 0.70

encodeMethod · 0.45

no test coverage detected