Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
chenpangpang
ComfyUI
Commits
bb663bcd
Commit
bb663bcd
authored
Jul 08, 2024
by
comfyanonymous
Browse files
Rename clip_t5base to t5base for stable audio text encoder.
parent
628f0b8e
Changes
2
Show whitespace changes
Inline
Side-by-side
Showing
2 changed files
with
10 additions
and
4 deletions
+10
-4
comfy/sa_t5.py
comfy/sa_t5.py
+1
-1
comfy/sd1_clip.py
comfy/sd1_clip.py
+9
-3
No files found.
comfy/sa_t5.py
View file @
bb663bcd
...
...
@@ -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
)
comfy/sd1_clip.py
View file @
bb663bcd
...
...
@@ -514,10 +514,16 @@ class SD1Tokenizer:
class
SD1ClipModel
(
torch
.
nn
.
Module
):
def
__init__
(
self
,
device
=
"cpu"
,
dtype
=
None
,
clip_name
=
"l"
,
clip_model
=
SDClipModel
,
**
kwargs
):
def
__init__
(
self
,
device
=
"cpu"
,
dtype
=
None
,
clip_name
=
"l"
,
clip_model
=
SDClipModel
,
name
=
None
,
**
kwargs
):
super
().
__init__
()
if
name
is
not
None
:
self
.
clip_name
=
name
self
.
clip
=
"{}"
.
format
(
self
.
clip_name
)
else
:
self
.
clip_name
=
clip_name
self
.
clip
=
"clip_{}"
.
format
(
self
.
clip_name
)
setattr
(
self
,
self
.
clip
,
clip_model
(
device
=
device
,
dtype
=
dtype
,
**
kwargs
))
self
.
dtypes
=
set
()
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment