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 torch import nn
from .ldm.modules.attention import CrossAttention from .ldm.modules.attention import CrossAttention
from inspect import isfunction from inspect import isfunction
import comfy.ops from comfy.ops import manual_cast
ops = comfy.ops.manual_cast ops = manual_cast
def exists(val): def exists(val):
return val is not None return val is not None

View File

@ -18,7 +18,7 @@
import torch import torch
import torch.nn as nn import torch.nn as nn
from comfy.ldm.modules.attention import optimized_attention from ..modules.attention import optimized_attention
class Linear(torch.nn.Linear): class Linear(torch.nn.Linear):
def reset_parameters(self): def reset_parameters(self):