From cb3923a90fd5f0cc01b4d9f58822465c8c68f8c9 Mon Sep 17 00:00:00 2001 From: Phil Wang Date: Wed, 13 Apr 2022 18:43:34 -0700 Subject: [PATCH] readme tweak --- README.md | 2 ++ 1 file changed, 2 insertions(+) diff --git a/README.md b/README.md index c0065a9..6ba0e09 100644 --- a/README.md +++ b/README.md @@ -142,6 +142,8 @@ Finally, the main contribution of the paper. The repository offers the diffusion import torch from dalle2_pytorch import DiffusionPriorNetwork, DiffusionPrior, CLIP +# get trained CLIP from step one + clip = CLIP( dim_text = 512, dim_image = 512,