wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-11 14:50:49 +08:00
Code Issues Actions 11 Packages Projects Releases Wiki Activity

Add xformer.ops imports

Browse Source
This commit is contained in:
Max Tretikov 2024-06-14 14:09:46 -06:00
parent ee44e3b1d7
commit 8b091f02de
3 changed files with 5 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
comfy/ldm/modules/attention.py
View File

@ -12,6 +12,7 @@ from ... import model_management
if model_management.xformers_enabled():
import xformers # pylint: disable=import-error
import xformers.ops # pylint: disable=import-error
from ...cli_args import args
from ... import ops

4
comfy/ldm/modules/diffusionmodules/model.py
View File

@ -11,8 +11,8 @@ from .... import ops
ops = ops.disable_weight_init
if model_management.xformers_enabled_vae():
import xformers
import xformers.ops
import xformers # pylint: disable=import-error
import xformers.ops # pylint: disable=import-error
def get_timestep_embedding(timesteps, embedding_dim):
"""

3
comfy/model_management.py
View File

@ -158,7 +158,8 @@ if args.disable_xformers:
XFORMERS_IS_AVAILABLE = False
else:
try:
import xformers # pylint: disable=import-error
import xformers # pylint: disable=import-error
import xformers.ops # pylint: disable=import-error
XFORMERS_IS_AVAILABLE = True
try: