Fix relative imports

This commit is contained in:
doctorpangloss 2024-02-21 13:31:55 -08:00
parent f419c5760d
commit 5cd06a727f
2 changed files with 3 additions and 3 deletions

View File

@ -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

View File

@ -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):