x = model.cond_stage_model.encode("a painting of a black cat with yellow eyes, a color pencil sketch by Betye Saar, deviantart, folk art, creative commons attribution, chalk art, charcoal drawing") y = torch.clamp((x + 1.0) / 2.0, min=0.0, max=1.0) Image.fromarray((255. * y[0]).cpu().numpy().astype(np.uint8)).save("token.png")