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

scaffold

Browse Source
This commit is contained in:
Phil Wang
2022-04-07 07:29:34 -07:00
parent 25fb133c83
commit f283bf25be
5 changed files with 80 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

1
dalle2_pytorch/__init__.py Normal file
View File

@@ -0,0 +1 @@
from dalle2_pytorch.dalle2_pytorch import DALLE2

11
dalle2_pytorch/dalle2_pytorch.py Normal file
View File

@@ -0,0 +1,11 @@
import torch
import torch.nn.functional as F
from torch import nn, einsum
from einops import rearrange
class DALLE2(nn.Module):
def __init__(self):
super().__init__()
def forward(self, x):
return x