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
4ba7fa02
Commit
4ba7fa02
authored
Jul 28, 2024
by
comfyanonymous
Browse files
Refactor: Move sd2_clip.py to text_encoders folder.
parent
ab76abc7
Changes
4
Hide whitespace changes
Inline
Side-by-side
Showing
4 changed files
with
5 additions
and
5 deletions
+5
-5
comfy/sd.py
comfy/sd.py
+3
-3
comfy/supported_models.py
comfy/supported_models.py
+2
-2
comfy/text_encoders/sd2_clip.py
comfy/text_encoders/sd2_clip.py
+0
-0
comfy/text_encoders/sd2_clip_config.json
comfy/text_encoders/sd2_clip_config.json
+0
-0
No files found.
comfy/sd.py
View file @
4ba7fa02
...
@@ -17,8 +17,8 @@ from . import diffusers_convert
...
@@ -17,8 +17,8 @@ from . import diffusers_convert
from
.
import
model_detection
from
.
import
model_detection
from
.
import
sd1_clip
from
.
import
sd1_clip
from
.
import
sd2_clip
from
.
import
sdxl_clip
from
.
import
sdxl_clip
import
comfy.text_encoders.sd2_clip
import
comfy.text_encoders.sd3_clip
import
comfy.text_encoders.sd3_clip
import
comfy.text_encoders.sa_t5
import
comfy.text_encoders.sa_t5
import
comfy.text_encoders.aura_t5
import
comfy.text_encoders.aura_t5
...
@@ -414,8 +414,8 @@ def load_clip(ckpt_paths, embedding_directory=None, clip_type=CLIPType.STABLE_DI
...
@@ -414,8 +414,8 @@ def load_clip(ckpt_paths, embedding_directory=None, clip_type=CLIPType.STABLE_DI
clip_target
.
clip
=
sdxl_clip
.
SDXLRefinerClipModel
clip_target
.
clip
=
sdxl_clip
.
SDXLRefinerClipModel
clip_target
.
tokenizer
=
sdxl_clip
.
SDXLTokenizer
clip_target
.
tokenizer
=
sdxl_clip
.
SDXLTokenizer
elif
"text_model.encoder.layers.22.mlp.fc1.weight"
in
clip_data
[
0
]:
elif
"text_model.encoder.layers.22.mlp.fc1.weight"
in
clip_data
[
0
]:
clip_target
.
clip
=
sd2_clip
.
SD2ClipModel
clip_target
.
clip
=
comfy
.
text_encoders
.
sd2_clip
.
SD2ClipModel
clip_target
.
tokenizer
=
sd2_clip
.
SD2Tokenizer
clip_target
.
tokenizer
=
comfy
.
text_encoders
.
sd2_clip
.
SD2Tokenizer
elif
"encoder.block.23.layer.1.DenseReluDense.wi_1.weight"
in
clip_data
[
0
]:
elif
"encoder.block.23.layer.1.DenseReluDense.wi_1.weight"
in
clip_data
[
0
]:
weight
=
clip_data
[
0
][
"encoder.block.23.layer.1.DenseReluDense.wi_1.weight"
]
weight
=
clip_data
[
0
][
"encoder.block.23.layer.1.DenseReluDense.wi_1.weight"
]
dtype_t5
=
weight
.
dtype
dtype_t5
=
weight
.
dtype
...
...
comfy/supported_models.py
View file @
4ba7fa02
...
@@ -3,8 +3,8 @@ from . import model_base
...
@@ -3,8 +3,8 @@ from . import model_base
from
.
import
utils
from
.
import
utils
from
.
import
sd1_clip
from
.
import
sd1_clip
from
.
import
sd2_clip
from
.
import
sdxl_clip
from
.
import
sdxl_clip
import
comfy.text_encoders.sd2_clip
import
comfy.text_encoders.sd3_clip
import
comfy.text_encoders.sd3_clip
import
comfy.text_encoders.sa_t5
import
comfy.text_encoders.sa_t5
import
comfy.text_encoders.aura_t5
import
comfy.text_encoders.aura_t5
...
@@ -101,7 +101,7 @@ class SD20(supported_models_base.BASE):
...
@@ -101,7 +101,7 @@ class SD20(supported_models_base.BASE):
return
state_dict
return
state_dict
def
clip_target
(
self
,
state_dict
=
{}):
def
clip_target
(
self
,
state_dict
=
{}):
return
supported_models_base
.
ClipTarget
(
sd2_clip
.
SD2Tokenizer
,
sd2_clip
.
SD2ClipModel
)
return
supported_models_base
.
ClipTarget
(
comfy
.
text_encoders
.
sd2_clip
.
SD2Tokenizer
,
comfy
.
text_encoders
.
sd2_clip
.
SD2ClipModel
)
class
SD21UnclipL
(
SD20
):
class
SD21UnclipL
(
SD20
):
unet_config
=
{
unet_config
=
{
...
...
comfy/sd2_clip.py
→
comfy/
text_encoders/
sd2_clip.py
View file @
4ba7fa02
File moved
comfy/sd2_clip_config.json
→
comfy/
text_encoders/
sd2_clip_config.json
View file @
4ba7fa02
File moved
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