added a wrapper model_manager_service and model events

This commit is contained in:
Lincoln Stein
2023-05-11 00:09:19 -04:00
parent fa6a580452
commit 4627910c5d
6 changed files with 495 additions and 8 deletions

View File

@ -59,7 +59,7 @@ class CompelInvocation(BaseInvocation):
# TODO: load without model
model = choose_model(context.services.model_manager, self.model)
pipeline = model["model"]
pipeline = model.context.model
tokenizer = pipeline.tokenizer
text_encoder = pipeline.text_encoder