From 55bdcb98b9ed78a358bab7cca63ec6555bd67b27 Mon Sep 17 00:00:00 2001
From: Phil Wang <lucidrains@gmail.com>
Date: Tue, 19 Apr 2022 09:26:58 -0700
Subject: [PATCH] scaffold for latent diffusion

---
 dalle2_pytorch/latent_diffusion.py | 12 ++++++++++++
 1 file changed, 12 insertions(+)
 create mode 100644 dalle2_pytorch/latent_diffusion.py

diff --git a/dalle2_pytorch/latent_diffusion.py b/dalle2_pytorch/latent_diffusion.py
new file mode 100644
index 0000000..3eaad3a
--- /dev/null
+++ b/dalle2_pytorch/latent_diffusion.py
@@ -0,0 +1,12 @@
+import torch
+import torch.nn.functional as F
+from torch import nn, einsum
+
+from einops import rearrange
+
+class LatentDiffusion(nn.Module):
+    def __init__(self):
+        super().__init__()
+
+    def forward(self, x):
+        return x