take care of DDPM decoder (DDPM for producing image embedding will have a separate objective, predicting directly the embedding rather than the noise [epsilon in paper])

This commit is contained in:
Phil Wang
2022-04-12 17:48:41 -07:00
parent 862e5ba50e
commit 33d69d3859
2 changed files with 172 additions and 12 deletions

View File

@@ -29,7 +29,7 @@ setup(
'torch>=1.10',
'torchvision',
'tqdm',
'x-clip>=0.4.3',
'x-clip>=0.4.4',
'youtokentome'
],
classifiers=[