Compare commits

...

1 Commits

Author SHA1 Message Date
github-actions[bot]
2f805a8122 chore(format): run black on dev 2024-01-14 07:42:36 +00:00
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(