refactor: merge Handler and Service layers into UseCase
This commit is contained in:
@@ -1,6 +1,6 @@
|
||||
from ._autoencoder_repository import AutoencoderRepository
|
||||
from ._pipeline_repository import PipelineRepository
|
||||
from ._sampling_repository import SamplingRepository
|
||||
from ._sampler_repository import SamplerRepository
|
||||
from ._scheduler_repository import SchedulerRepository
|
||||
from ._text_encoder_repository import TextEncoderRepository
|
||||
|
||||
@@ -8,6 +8,6 @@ __all__ = [
|
||||
"PipelineRepository",
|
||||
"AutoencoderRepository",
|
||||
"TextEncoderRepository",
|
||||
"SamplingRepository",
|
||||
"SamplerRepository",
|
||||
"SchedulerRepository",
|
||||
]
|
||||
|
||||
+1
-1
@@ -6,7 +6,7 @@ from diffusers.schedulers.scheduling_utils import SchedulerMixin
|
||||
from PIL import Image
|
||||
|
||||
|
||||
class SamplingRepository(ABC):
|
||||
class SamplerRepository(ABC):
|
||||
@abstractmethod
|
||||
def sample(
|
||||
self,
|
||||
Reference in New Issue
Block a user