fix lint error

This commit is contained in:
Moeblack 2023-06-10 16:59:14 +08:00 committed by henryruhs
parent d603778f67
commit 60c0436c61

View File

@ -45,6 +45,7 @@ else:
ckpt_path = "CodeFormer/weights/CodeFormer/codeformer.pth"
checkpoint = torch.load(ckpt_path)["params_ema"]
def get_code_former():
global CODE_FORMER
with THREAD_LOCK: