diff --git a/modules/models/sd3/sd3_model.py b/modules/models/sd3/sd3_model.py index 98470cdab..dbec8168f 100644 --- a/modules/models/sd3/sd3_model.py +++ b/modules/models/sd3/sd3_model.py @@ -73,7 +73,7 @@ class SD3Inferencer(torch.nn.Module): def medvram_fields(self): return [ (self, 'first_stage_model'), - (self, 'cond_stage_model'), + (self, 'text_encoders'), (self, 'model'), ]