Skip to content

Commit 4ba7fa0

Browse files
Refactor: Move sd2_clip.py to text_encoders folder.
1 parent ab76abc commit 4ba7fa0

File tree

4 files changed

+5
-5
lines changed

4 files changed

+5
-5
lines changed

comfy/sd.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff line change
@@ -17,8 +17,8 @@
1717
from . import model_detection
1818

1919
from . import sd1_clip
20-
from . import sd2_clip
2120
from . import sdxl_clip
21+
import comfy.text_encoders.sd2_clip
2222
import comfy.text_encoders.sd3_clip
2323
import comfy.text_encoders.sa_t5
2424
import comfy.text_encoders.aura_t5
@@ -414,8 +414,8 @@ class EmptyClass:
414414
clip_target.clip = sdxl_clip.SDXLRefinerClipModel
415415
clip_target.tokenizer = sdxl_clip.SDXLTokenizer
416416
elif "text_model.encoder.layers.22.mlp.fc1.weight" in clip_data[0]:
417-
clip_target.clip = sd2_clip.SD2ClipModel
418-
clip_target.tokenizer = sd2_clip.SD2Tokenizer
417+
clip_target.clip = comfy.text_encoders.sd2_clip.SD2ClipModel
418+
clip_target.tokenizer = comfy.text_encoders.sd2_clip.SD2Tokenizer
419419
elif "encoder.block.23.layer.1.DenseReluDense.wi_1.weight" in clip_data[0]:
420420
weight = clip_data[0]["encoder.block.23.layer.1.DenseReluDense.wi_1.weight"]
421421
dtype_t5 = weight.dtype

comfy/supported_models.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -3,8 +3,8 @@
33
from . import utils
44

55
from . import sd1_clip
6-
from . import sd2_clip
76
from . import sdxl_clip
7+
import comfy.text_encoders.sd2_clip
88
import comfy.text_encoders.sd3_clip
99
import comfy.text_encoders.sa_t5
1010
import comfy.text_encoders.aura_t5
@@ -101,7 +101,7 @@ def process_clip_state_dict_for_saving(self, state_dict):
101101
return state_dict
102102

103103
def clip_target(self, state_dict={}):
104-
return supported_models_base.ClipTarget(sd2_clip.SD2Tokenizer, sd2_clip.SD2ClipModel)
104+
return supported_models_base.ClipTarget(comfy.text_encoders.sd2_clip.SD2Tokenizer, comfy.text_encoders.sd2_clip.SD2ClipModel)
105105

106106
class SD21UnclipL(SD20):
107107
unet_config = {
File renamed without changes.
File renamed without changes.

0 commit comments

Comments
 (0)