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

more accurate readme

Browse Source
This commit is contained in:
Phil Wang
2022-07-09 20:57:26 -07:00
committed by GitHub
parent 3dae43fa0e
commit 4173e88121
1 changed files with 2 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

3
README.md
View File

@@ -355,7 +355,8 @@ prior_network = DiffusionPriorNetwork(
diffusion_prior = DiffusionPrior( diffusion_prior = DiffusionPrior(
net = prior_network, net = prior_network,
clip = clip, clip = clip,
timesteps = 100, timesteps = 1000,
sample_timesteps = 64,
cond_drop_prob = 0.2 cond_drop_prob = 0.2
).cuda() ).cuda()