Method _encode

lwm/vqgan.py:33–40

(self)

    
    @cached_property
    def _encode(self):
        def fn(pixel_values, params):
            return self.model.apply(
                {'params': params}, 
                pixel_values,
                method=self.model.encode
            )
        return partial(self._wrap_fn(fn), params=self.params)
    
    @cached_property
    def _decode(self):