aljaz/DALLE2-pytorch
1
1
Fork 0
mirror of https://github.com/lucidrains/DALLE2-pytorch.git synced 2025-12-23 19:44:20 +01:00
Code Issues Packages Projects Releases Wiki Activity

readme tweak

Browse Source
This commit is contained in:
Phil Wang
2022-04-13 18:43:34 -07:00
committed by GitHub
parent cc30676a3f
commit cb3923a90f
1 changed files with 2 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
README.md
View File

@@ -142,6 +142,8 @@ Finally, the main contribution of the paper. The repository offers the diffusion
import torch import torch
from dalle2_pytorch import DiffusionPriorNetwork, DiffusionPrior, CLIP from dalle2_pytorch import DiffusionPriorNetwork, DiffusionPrior, CLIP
# get trained CLIP from step one
clip = CLIP( clip = CLIP(
dim_text = 512, dim_text = 512,
dim_image = 512, dim_image = 512,