mirror of
https://github.com/vladmandic/automatic
synced 2026-08-27 23:51:00 +02:00
Vladimir Mandic
25 lines
796 B
Python
25 lines
796 B
Python
import sys
|
|
from modules import shared
|
|
|
|
|
|
def get_loader(component):
|
|
"""Return loader type for log messages."""
|
|
if sys.platform != 'linux':
|
|
return 'default'
|
|
if component == 'diffusers':
|
|
return 'runai' if shared.opts.runai_streamer_diffusers else 'default'
|
|
return 'runai' if shared.opts.runai_streamer_transformers else 'default'
|
|
|
|
|
|
def set_pipeline(name, cls):
|
|
"""Set pipeline class in shared_items.pipelines with logging."""
|
|
import diffusers
|
|
from modules.logger import log
|
|
from modules import shared_items
|
|
if shared_items.pipelines.get(name, None) is not None:
|
|
return
|
|
shared_items.pipelines[name] = cls
|
|
setattr(diffusers, name, cls)
|
|
setattr(diffusers.pipelines, name, cls)
|
|
log.debug(f'Pipeline: {name}={cls} initialized')
|