Mirrors
/
diffusers
mirror of https://github.com/ShivamShrirao/diffusers.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

make style

This commit is contained in:
Patrick von Platen 2022-12-20 01:38:45 +01:00
parent 8e4733b3c3
commit e29dc97215
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
src/diffusers/models/attention.py
View File

@ -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):
) )
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: