Rename clip_t5base to t5base for stable audio text encoder.

2024-07-08 08:48:38 -04:00
parent 628f0b8ebc
commit bb663bcd6c
2 changed files with 10 additions and 4 deletions
--- a/comfy/sa_t5.py
+++ b/comfy/sa_t5.py
@@ -19,4 +19,4 @@ class SAT5Tokenizer(sd1_clip.SD1Tokenizer):

 class SAT5Model(sd1_clip.SD1ClipModel):
    def __init__(self, device="cpu", dtype=None, **kwargs):
-        super().__init__(device=device, dtype=dtype, clip_name="t5base", clip_model=T5BaseModel, **kwargs)
+        super().__init__(device=device, dtype=dtype, name="t5base", clip_model=T5BaseModel, **kwargs)