Replace with f-string

This commit is contained in:
Ftps
2023-11-17 18:23:39 +09:00
parent a8fbbdad18
commit 7b5ec6eb72

View File

@@ -131,11 +131,11 @@ class Config:
or "1070" in self.gpu_name
or "1080" in self.gpu_name
):
logger.info("Found GPU %s, force to fp32", self.gpu_name)
logger.info(f"Found GPU {self.gpu_name}, force to fp32")
self.is_half = False
self.use_fp32_config()
else:
logger.info("Found GPU %s", self.gpu_name)
logger.info(f"Found GPU {self.gpu_name}")
self.gpu_mem = int(
torch.cuda.get_device_properties(i_device).total_memory
/ 1024