fix --medvram

2024-06-30 07:06:28 +03:00 · 2024-06-30 07:06:28 +03:00 · 9e404c3154
parent ebe8be9028
commit 9e404c3154
1 changed files with 1 additions and 1 deletions
--- 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'),
        ]