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

rename to DiffusionPriorNetwork in case ARPriorNetwork is ever built

Browse Source
This commit is contained in:
Phil Wang
2022-04-12 11:45:57 -07:00
parent fd38eb83c4
commit 40aa304b7e
1 changed files with 1 additions and 1 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
dalle2_pytorch/dalle2_pytorch.py
View File

@@ -149,7 +149,7 @@ class Transformer(nn.Module):
return self.norm(x) return self.norm(x)
class PriorNetwork(nn.Module): class DiffusionPriorNetwork(nn.Module):
def __init__( def __init__(
self, self,
dim, dim,