chore(format): run black on dev (#1717)

Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
This commit is contained in:
github-actions[bot] 2024-01-14 16:43:14 +09:00 committed by GitHub
parent bb4d185a57
commit 4f7b6b2b9e
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
3 changed files with 13 additions and 8 deletions

View File

@ -158,7 +158,8 @@ class Config:
/ 1024
+ 0.4
)
if self.gpu_mem <= 4: self.preprocess_per = 3.0
if self.gpu_mem <= 4:
self.preprocess_per = 3.0
elif self.has_mps():
logger.info("No supported Nvidia GPU found")
self.device = self.instead = "mps"
@ -241,5 +242,8 @@ class Config:
)
except:
pass
logger.info("Half-precision floating-point: %s, device: %s" % (self.is_half, self.device))
logger.info(
"Half-precision floating-point: %s, device: %s"
% (self.is_half, self.device)
)
return x_pad, x_query, x_center, x_max

View File

@ -31,6 +31,7 @@ def println(strr):
f.write("%s\n" % strr)
f.flush()
class PreProcess:
def __init__(self, sr, exp_dir, per=3.7):
self.slicer = Slicer(