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
2023-01-07 08:41:06 -08:00
parent 8dbf76268d
commit 2e744ef2f2
1 changed files with 0 additions and 1 deletions
@@ -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)],