change print

HNR1 · Jun 13, 2023 · 99035ff · 99035ff
1 parent 50772b6
commit 99035ff
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/tomesd/patch.py b/tomesd/patch.py
@@ -77,7 +77,7 @@ def make_diffusers_tome_block(block_class: Type[torch.nn.Module]) -> Type[torch.
     Make a patched class for a diffusers model.
     This patch applies ToMe to the forward function of the block.
     """
-    print(type(block_class))
+    print(block_class)
     class ToMeBlock(block_class):
         # Save for unpatching later
         _parent = block_class