mirror of
https://github.com/comfyanonymous/ComfyUI.git
synced 2026-01-11 23:00:51 +08:00
b0be335d59
- ComfyUI can now load EXR files. - There are new arithmetic nodes for floats and integers. - EXR nodes can load depth maps and be remapped with ImageApplyColormap. This allows end users to use ground truth depth data from video game engines or 3D graphics tools and recolor it to the format expected by depth ControlNets: grayscale inverse depth maps and "inferno" colored inverse depth maps. - Fixed license notes. - Added an additional known ControlNet model. - Because CV2 is now used to read OpenEXR files, an environment variable must be set early on in the application, before CV2 is imported. This file, main_pre, is now imported early on in more places.
36 lines
1.3 KiB
Python
36 lines
1.3 KiB
Python
"""
|
|
This should be imported before entrypoints to correctly configure global options prior to importing packages like torch and cv2.
|
|
|
|
Use this instead of cli_args to import the args:
|
|
|
|
>>> from comfy.cmd.main_pre import args
|
|
|
|
It will enable command line argument parsing. If this isn't desired, you must author your own implementation of these fixes.
|
|
"""
|
|
import os
|
|
|
|
from .. import options
|
|
|
|
import warnings
|
|
import logging
|
|
|
|
options.enable_args_parsing()
|
|
if os.name == "nt":
|
|
logging.getLogger("xformers").addFilter(lambda record: 'A matching Triton is not available' not in record.getMessage())
|
|
warnings.filterwarnings("ignore", message="torch.utils._pytree._register_pytree_node is deprecated. Please use torch.utils._pytree.register_pytree_node instead.")
|
|
warnings.filterwarnings("ignore", message="Torch was not compiled with flash attention.")
|
|
warnings.filterwarnings("ignore", message=".*Torch was not compiled with flash attention.*")
|
|
|
|
from ..cli_args import args
|
|
|
|
if args.cuda_device is not None:
|
|
os.environ['CUDA_VISIBLE_DEVICES'] = str(args.cuda_device)
|
|
logging.info("Set cuda device to:", args.cuda_device)
|
|
|
|
if args.deterministic:
|
|
if 'CUBLAS_WORKSPACE_CONFIG' not in os.environ:
|
|
os.environ['CUBLAS_WORKSPACE_CONFIG'] = ":4096:8"
|
|
|
|
os.environ["OPENCV_IO_ENABLE_OPENEXR"] = "1"
|
|
__all__ = ["args"]
|