Method get_sample_input

demo/Diffusion/models.py:373–375

(self, batch_size, image_height, image_width)

        }

    def get_sample_input(self, batch_size, image_height, image_width):
        latent_height, latent_width = self.check_dims(batch_size, image_height, image_width)
        return torch.randn(batch_size, 4, latent_height, latent_width, dtype=torch.float32, device=self.device)

def make_VAE(version, hf_token, device, verbose, max_batch_size, inpaint=False):
    return VAE(hf_token=hf_token, device=device, verbose=verbose, path=get_path(version, inpaint=inpaint),