Skip to content

GitLab

Commit e29dc972 authored by Patrick von Platen's avatar Patrick von Platen
Browse files

make style

parent 8e4733b3
Hide whitespace changes
Inline Side-by-side
Showing with 4 additions and 4 deletions
src/diffusers/models/attention.py
View file @ e29dc972
......@@ -297,8 +297,8 @@ class AttentionBlock(nn.Module):
)
elif not torch.cuda.is_available():
raise ValueError(
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is only"
" available for GPU "
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is"
" only available for GPU "
)
else:
try:
......@@ -461,8 +461,8 @@ class BasicTransformerBlock(nn.Module):
)
elif not torch.cuda.is_available():
raise ValueError(
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is only"
" available for GPU "
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is"
" only available for GPU "
)
else:
try:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment