From 5cd06a727fb89581677e325e11de8898beea01c9 Mon Sep 17 00:00:00 2001 From: doctorpangloss <@hiddenswitch.com> Date: Wed, 21 Feb 2024 13:31:55 -0800 Subject: [PATCH] Fix relative imports --- comfy/gligen.py | 4 ++-- comfy/ldm/cascade/common.py | 2 +- 2 files changed, 3 insertions(+), 3 deletions(-) diff --git a/comfy/gligen.py b/comfy/gligen.py index 85753e890..2c3a645b1 100644 --- a/comfy/gligen.py +++ b/comfy/gligen.py @@ -4,8 +4,8 @@ import torch from torch import nn from .ldm.modules.attention import CrossAttention from inspect import isfunction -import comfy.ops -ops = comfy.ops.manual_cast +from comfy.ops import manual_cast +ops = manual_cast def exists(val): return val is not None diff --git a/comfy/ldm/cascade/common.py b/comfy/ldm/cascade/common.py index 124902c09..f7092a7ad 100644 --- a/comfy/ldm/cascade/common.py +++ b/comfy/ldm/cascade/common.py @@ -18,7 +18,7 @@ import torch import torch.nn as nn -from comfy.ldm.modules.attention import optimized_attention +from ..modules.attention import optimized_attention class Linear(torch.nn.Linear): def reset_parameters(self):