Skip to content

Commit e9fd0ca

Browse files
committed
update
1 parent 6c15a10 commit e9fd0ca

File tree

1 file changed

+2
-1
lines changed

1 file changed

+2
-1
lines changed

src/diffusers/models/attention_dispatch.py

Lines changed: 2 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -179,7 +179,6 @@ def decorator(func):
179179

180180
@classmethod
181181
def get_active_backend(cls):
182-
_check_backend_requirements(cls._active_backend)
183182
return cls._active_backend, cls._backends[cls._active_backend]
184183

185184
@classmethod
@@ -227,6 +226,8 @@ def dispatch_attention_fn(
227226
backend_name = AttentionBackendName(backend)
228227
backend_fn = _AttentionBackendRegistry._backends.get(backend_name)
229228

229+
_check_backend_requirements(backend_name)
230+
230231
kwargs = {
231232
"query": query,
232233
"key": key,

0 commit comments

Comments
 (0)