Mirrored_Repos/InvokeAI
1
0
Fork 0
mirror of https://github.com/invoke-ai/InvokeAI synced 2024-08-30 20:32:17 +00:00
Code Issues Packages Projects Releases Wiki Activity

Apply black, isort, flake8

Browse Source
This commit is contained in:
Martin Kristiansen
2023-09-11 09:57:41 -04:00
committed by Kent Keirsey
parent 0450c28f14
commit e467ca7f1b
14 changed files with 32 additions and 57 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
invokeai/backend/util/attention.py
View File

@ -3,8 +3,8 @@
Utility routine used for autodetection of optimal slice size
for attention mechanism.
"""
import torch
import psutil
import torch
def auto_detect_slice_size(latents: torch.Tensor) -> str: