From a18a6a0773da496cadbb2ec291cb9ea4ea63775d Mon Sep 17 00:00:00 2001 From: George Hotz Date: Tue, 6 Sep 2022 16:51:08 -0700 Subject: [PATCH] fix sd with TORCH=1 --- examples/stable_diffusion.py | 5 ++--- 1 file changed, 2 insertions(+), 3 deletions(-) diff --git a/examples/stable_diffusion.py b/examples/stable_diffusion.py index 6c89dc6c19..be2dd46b97 100644 --- a/examples/stable_diffusion.py +++ b/examples/stable_diffusion.py @@ -10,8 +10,8 @@ from collections import namedtuple from extra.utils import fake_torch_load_zipped, get_child from tinygrad.nn import Conv2d from tinygrad.tensor import Tensor -from tinygrad.helpers import prod -from extra.introspection import print_objects + +# TODO: refactor AttnBlock, CrossAttention, CLIPAttention to share code # TODO: rename to GroupNorm and put in nn.py class Normalize: @@ -560,7 +560,6 @@ if __name__ == "__main__": def get_model_output(latent, t): # put into diffuser timesteps = Tensor([t]) - print_objects() unconditional_latent = model.model.diffusion_model(latent, timesteps, unconditional_context).realize() latent = model.model.diffusion_model(latent, timesteps, context).realize()