From 2e744ef2f2d02a8f5bfb94ceb566e3af477bdae0 Mon Sep 17 00:00:00 2001
From: nogira <83847986+nogira@users.noreply.github.com>
Date: Sun, 8 Jan 2023 03:41:06 +1100
Subject: [PATCH] confirmed (#449)

w/ a bunch of print statements in the official model here: https://github.com/CompVis/stable-diffusion/blob/ce05de28194041e030ccfc70c635fe3707cdfc30/ldm/modules/diffusionmodules/openaimodel.py#L413
---
 examples/stable_diffusion.py | 1 -
 1 file changed, 1 deletion(-)

diff --git a/examples/stable_diffusion.py b/examples/stable_diffusion.py
index 9a30c5a76c..3a58316039 100644
--- a/examples/stable_diffusion.py
+++ b/examples/stable_diffusion.py
@@ -310,7 +310,6 @@ class UNetModel:
     ]
     self.input_blocks = [
       [Conv2d(4, 320, kernel_size=3, padding=1)],
-      # TODO: my head sizes and counts are a guess
       [ResBlock(320, 1280, 320), SpatialTransformer(320, 768, 8, 40)],
       [ResBlock(320, 1280, 320), SpatialTransformer(320, 768, 8, 40)],
       [Downsample(320)],