Public/automatic
1
0
Fork 0
mirror of https://github.com/vladmandic/automatic synced 2026-09-20 01:31:13 +02:00
Code Issues Packages Projects Releases Wiki Activity

unify hf-login

Browse Source
This commit is contained in:
Vladimir Mandic
2024-06-15 16:47:03 -04:00
parent 6c1d59d367
commit 5feaed0c6a
4 changed files with 26 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
modules/sd_models.py
+2 -1
View File
@@ -1381,7 +1381,8 @@ def set_diffusers_attention(pipe):
modules = [getattr(pipe, n, None) for n in module_names]
modules = [m for m in modules if isinstance(m, torch.nn.Module) and hasattr(m, "set_attn_processor")]
for module in modules:
if 'SD3Transformer2DModel' in module.__class__.__name__: # TODO SD3
print('HERE', module.__class__.__name__)
if module.__class__.__name__ in ['SD3Transformer2DModel']:
module.set_attn_processor(p.JointAttnProcessor2_0())
else:
module.set_attn_processor(attention)