fix issue with embeddings being loaded twice

- as noted by JPPhoto
This commit is contained in:
Lincoln Stein 2023-03-25 10:45:03 -04:00
parent 9536ba22af
commit 5ac0316c62
2 changed files with 3 additions and 5 deletions

View File

@ -434,7 +434,6 @@ class ModelManager(object):
height = width height = width
print(f" | Default image dimensions = {width} x {height}") print(f" | Default image dimensions = {width} x {height}")
self._add_embeddings_to_model(pipeline)
return pipeline, width, height, model_hash return pipeline, width, height, model_hash
def _load_ckpt_model(self, model_name, mconfig): def _load_ckpt_model(self, model_name, mconfig):
@ -476,7 +475,6 @@ class ModelManager(object):
pipeline.enable_offload_submodels(self.device) pipeline.enable_offload_submodels(self.device)
else: else:
pipeline.to(self.device) pipeline.to(self.device)
return ( return (
pipeline, pipeline,
width, width,