readme tweak

This commit is contained in:
Phil Wang
2022-04-13 18:43:34 -07:00
committed by GitHub
parent cc30676a3f
commit cb3923a90f

View File

@@ -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,