mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-27 14:50:20 +08:00
Fix relative imports
This commit is contained in:
parent
f419c5760d
commit
5cd06a727f
@ -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
|
||||||
|
|||||||
@ -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):
|
||||||
|
|||||||
Loading…
Reference in New Issue
Block a user