mirrors/tinygrad
2
0
Fork 0
mirror of https://github.com/tinygrad/tinygrad.git synced 2026-06-24 02:14:17 +00:00
Code Issues Projects Releases Packages Wiki Activity

confirmed (#449)

Browse source 
w/ a bunch of print statements in the official model here: ce05de2819/ldm/modules/diffusionmodules/openaimodel.py (L413)
This commit is contained in:
nogira 2023-01-08 03:41:06 +11:00 • committed by GitHub
commit 2e744ef2f2
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 0 additions and 1 deletions
Download patch file Download diff file Expand all files Collapse all files

1
examples/stable_diffusion.py
View file

@ -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)],