mirror of
https://github.com/invoke-ai/InvokeAI
synced 2024-08-30 20:32:17 +00:00
d176fb07cd
Allowed values are 'auto', 'float32', 'autocast', 'float16'. If not specified or 'auto' a working precision is automatically selected based on the torch device. Context: #526 Deprecated --full_precision / -F Tested on both cuda and cpu by calling scripts/dream.py without arguments and checked the auto configuration worked. With --precision=auto/float32/autocast/float16 it performs as expected, either working or failing with a reasonable error. Also checked Img2Img.
82 lines
2.3 KiB
Python
82 lines
2.3 KiB
Python
# Copyright (c) 2022 Kyle Schouviller (https://github.com/kyle0654)
|
|
|
|
"""Containers module."""
|
|
|
|
from dependency_injector import containers, providers
|
|
from flask_socketio import SocketIO
|
|
from ldm.generate import Generate
|
|
from server import services
|
|
|
|
class Container(containers.DeclarativeContainer):
|
|
wiring_config = containers.WiringConfiguration(packages=['server'])
|
|
|
|
config = providers.Configuration()
|
|
|
|
socketio = providers.ThreadSafeSingleton(
|
|
SocketIO,
|
|
app = None
|
|
)
|
|
|
|
# TODO: Add a model provider service that provides model(s) dynamically
|
|
model_singleton = providers.ThreadSafeSingleton(
|
|
Generate,
|
|
model = config.model,
|
|
sampler_name = config.sampler_name,
|
|
embedding_path = config.embedding_path,
|
|
precision = config.precision
|
|
# config = config.model.config,
|
|
|
|
# width = config.model.width,
|
|
# height = config.model.height,
|
|
# sampler_name = config.model.sampler_name,
|
|
# weights = config.model.weights,
|
|
# precision = config.model.precision,
|
|
# grid = config.model.grid,
|
|
# seamless = config.model.seamless,
|
|
# embedding_path = config.model.embedding_path,
|
|
# device_type = config.model.device_type
|
|
)
|
|
|
|
# TODO: get location from config
|
|
image_storage_service = providers.ThreadSafeSingleton(
|
|
services.ImageStorageService,
|
|
'./outputs/img-samples/'
|
|
)
|
|
|
|
# TODO: get location from config
|
|
image_intermediates_storage_service = providers.ThreadSafeSingleton(
|
|
services.ImageStorageService,
|
|
'./outputs/intermediates/'
|
|
)
|
|
|
|
signal_queue_service = providers.ThreadSafeSingleton(
|
|
services.SignalQueueService
|
|
)
|
|
|
|
signal_service = providers.ThreadSafeSingleton(
|
|
services.SignalService,
|
|
socketio = socketio,
|
|
queue = signal_queue_service
|
|
)
|
|
|
|
generation_queue_service = providers.ThreadSafeSingleton(
|
|
services.JobQueueService
|
|
)
|
|
|
|
# TODO: get locations from config
|
|
log_service = providers.ThreadSafeSingleton(
|
|
services.LogService,
|
|
'./outputs/img-samples/',
|
|
'dream_web_log.txt'
|
|
)
|
|
|
|
generator_service = providers.ThreadSafeSingleton(
|
|
services.GeneratorService,
|
|
model = model_singleton,
|
|
queue = generation_queue_service,
|
|
imageStorage = image_storage_service,
|
|
intermediateStorage = image_intermediates_storage_service,
|
|
log = log_service,
|
|
signal_service = signal_service
|
|
)
|