automatically detect v-parameterization for SD2 checkpoints

This commit is contained in:
AUTOMATIC
2023-01-28 15:24:29 +03:00
parent 4aa7f5b5b9
commit d04e3e921e
2 changed files with 48 additions and 5 deletions

View File

@@ -131,6 +131,8 @@ class StableDiffusionModelHijack:
m.cond_stage_model.wrapped.model.token_embedding = m.cond_stage_model.wrapped.model.token_embedding.wrapped
m.cond_stage_model = m.cond_stage_model.wrapped
undo_optimizations()
self.apply_circular(False)
self.layers = None
self.clip = None