Unload sd_model before loading the other

This commit is contained in:
Jairo Correa
2022-11-01 04:01:49 -03:00
parent 5c9b3625fa
commit af758e97fa
5 changed files with 34 additions and 10 deletions

View File

@@ -94,6 +94,10 @@ class StableDiffusionModelHijack:
if type(model_embeddings.token_embedding) == EmbeddingsWithFixes:
model_embeddings.token_embedding = model_embeddings.token_embedding.wrapped
self.layers = None
self.circular_enabled = False
self.clip = None
def apply_circular(self, enable):
if self.circular_enabled == enable:
return