wangbo/ComfyUI
1
0
Fork 0
mirror of https://github.com/comfyanonymous/ComfyUI.git synced 2026-01-23 12:50:18 +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(): if model_management.xformers_enabled():
import xformers # pylint: disable=import-error import xformers # pylint: disable=import-error
import xformers.ops # pylint: disable=import-error
from ...cli_args import args from ...cli_args import args
from ... import ops from ... import ops

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

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

1
comfy/model_management.py
View File

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