From 25e97a889586d0a69a138ef3b6351d1f6f65db3e Mon Sep 17 00:00:00 2001 From: layerdiffusion <19834515+lllyasviel@users.noreply.github.com> Date: Tue, 20 Aug 2024 21:27:16 -0700 Subject: [PATCH] revise space --- spaces.py | 1 + 1 file changed, 1 insertion(+) diff --git a/spaces.py b/spaces.py index dc0d56ce..a4ef7aff 100644 --- a/spaces.py +++ b/spaces.py @@ -179,6 +179,7 @@ def automatically_move_to_gpu_when_forward(m: torch.nn.Module, target_model: tor m.forge_space_hooked_names = [] if method_name in m.forge_space_hooked_names: + print(f'Already hooked {type(m).__name__}.{method_name}') return print(f'Automatic hook: {type(m).__name__}.{method_name}')