Skip to content

Commit 056f81c

Browse files
authored
fix wrong param and typo (#105)
1 parent e7e92b4 commit 056f81c

File tree

5 files changed

+5
-5
lines changed

5 files changed

+5
-5
lines changed

diffsynth_engine/models/basic/attention.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -62,7 +62,7 @@ def sage_attn(q, k, v, attn_mask=None, scale=None):
6262
if SPARGE_ATTN_AVAILABLE:
6363
from spas_sage_attn import spas_sage2_attn_meansim_cuda
6464

65-
def sparge_attn(self, q, k, v, attn_mask=None, scale=None):
65+
def sparge_attn(q, k, v, attn_mask=None, scale=None):
6666
q = q.transpose(1, 2)
6767
k = k.transpose(1, 2)
6868
v = v.transpose(1, 2)

diffsynth_engine/models/flux/flux_dit.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -156,7 +156,7 @@ def convert(self, state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor
156156
state_dict = self._from_diffusers(state_dict)
157157
logger.info("use diffusers format state dict")
158158
else:
159-
logger.info("user diffsynth format state dict")
159+
logger.info("use diffsynth format state dict")
160160
return state_dict
161161

162162

diffsynth_engine/models/flux/flux_text_encoder.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -37,7 +37,7 @@ def convert(self, state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor
3737
state_dict = self._from_diffusers(state_dict)
3838
logger.info("use diffusers format state dict")
3939
else:
40-
logger.info("user diffsynth format state dict")
40+
logger.info("use diffsynth format state dict")
4141
return state_dict
4242

4343

diffsynth_engine/models/sd/sd_unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -167,7 +167,7 @@ def convert(self, state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor
167167
state_dict = self._from_diffusers(state_dict)
168168
logger.info("use diffusers format state dict")
169169
else:
170-
logger.info("user diffsynth format state dict")
170+
logger.info("use diffsynth format state dict")
171171
return state_dict
172172

173173

diffsynth_engine/models/sdxl/sdxl_unet.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -154,7 +154,7 @@ def convert(self, state_dict: Dict[str, torch.Tensor]) -> Dict[str, torch.Tensor
154154
state_dict = self._from_diffusers(state_dict)
155155
logger.info("use diffusers format state dict")
156156
else:
157-
logger.info("user diffsynth format state dict")
157+
logger.info("use diffsynth format state dict")
158158
return state_dict
159159

160160

0 commit comments

Comments
 (0)