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): ...@@ -297,8 +297,8 @@ class AttentionBlock(nn.Module):
) )
elif not torch.cuda.is_available(): elif not torch.cuda.is_available():
raise ValueError( raise ValueError(
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is only" "torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is"
" available for GPU " " only available for GPU "
) )
else: else:
try: try:
...@@ -461,8 +461,8 @@ class BasicTransformerBlock(nn.Module): ...@@ -461,8 +461,8 @@ class BasicTransformerBlock(nn.Module):
) )
elif not torch.cuda.is_available(): elif not torch.cuda.is_available():
raise ValueError( raise ValueError(
"torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is only" "torch.cuda.is_available() should be True but is False. xformers' memory efficient attention is"
" available for GPU " " only available for GPU "
) )
else: else:
try: 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