From 02835566084b51cb93b03b7acf2281c5773fd654 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Fri, 29 Apr 2022 13:40:55 -0700 Subject: [PATCH] fix example in readme, since api changed --- README.md | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/README.md b/README.md index e4fee0b..6b809e6 100644 --- a/README.md +++ b/README.md @@ -430,8 +430,8 @@ images = torch.randn(4, 3, 256, 256).cuda() # precompute the text and image embeddings # here using the diffusion prior class, but could be done with CLIP alone -clip_image_embeds = diffusion_prior.get_image_embed(images) -clip_text_embeds = diffusion_prior.get_text_cond(text).get('text_embed') +clip_image_embeds = diffusion_prior.clip.embed_image(images).image_embed +clip_text_embeds = diffusion_prior.clip.embed_text(text).text_embed # feed text and images into diffusion prior network