diff --git a/comfy/ops.py b/comfy/ops.py index f6f85de60..f674b47f7 100644 --- a/comfy/ops.py +++ b/comfy/ops.py @@ -1,5 +1,4 @@ import torch -from contextlib import contextmanager import comfy.model_management def cast_bias_weight(s, input):