mirror of
https://github.com/lucidrains/DALLE2-pytorch.git
synced 2026-02-23 02:14:28 +01:00
Compare commits
2 Commits
| Author | SHA1 | Date | |
|---|---|---|---|
|
|
cc58f75474 | ||
|
|
3b2cf7b0bc |
@@ -1124,7 +1124,7 @@ class DiffusionPriorNetwork(nn.Module):
|
|||||||
learned_queries = repeat(self.learned_query, 'd -> b 1 d', b = batch)
|
learned_queries = repeat(self.learned_query, 'd -> b 1 d', b = batch)
|
||||||
|
|
||||||
if self.self_cond:
|
if self.self_cond:
|
||||||
learned_queries = torch.cat((image_embed, self_cond), dim = -2)
|
learned_queries = torch.cat((self_cond, learned_queries), dim = -2)
|
||||||
|
|
||||||
tokens = torch.cat((
|
tokens = torch.cat((
|
||||||
text_encodings,
|
text_encodings,
|
||||||
|
|||||||
@@ -1 +1 @@
|
|||||||
__version__ = '1.12.0'
|
__version__ = '1.12.2'
|
||||||
|
|||||||
2
setup.py
2
setup.py
@@ -27,7 +27,7 @@ setup(
|
|||||||
'accelerate',
|
'accelerate',
|
||||||
'click',
|
'click',
|
||||||
'open-clip-torch>=2.0.0,<3.0.0',
|
'open-clip-torch>=2.0.0,<3.0.0',
|
||||||
'clip-anytorch>=2.4.0',
|
'clip-anytorch>=2.5.2',
|
||||||
'coca-pytorch>=0.0.5',
|
'coca-pytorch>=0.0.5',
|
||||||
'ema-pytorch>=0.0.7',
|
'ema-pytorch>=0.0.7',
|
||||||
'einops>=0.4',
|
'einops>=0.4',
|
||||||
|
|||||||
Reference in New Issue
Block a user