fix: requriements & config

This commit is contained in:
源文雨
2023-08-30 00:23:24 +08:00
parent 47a3882b3a
commit 417c1e58a1
6 changed files with 229 additions and 219 deletions

View File

@@ -23,7 +23,15 @@ def use_fp32_config():
with open("infer/modules/train/preprocess.py", "w") as f:
f.write(strr)
def singleton_variable(func):
def wrapper(*args, **kwargs):
if not wrapper.instance:
wrapper.instance = func(*args, **kwargs)
return wrapper.instance
wrapper.instance = None
return wrapper
@singleton_variable
class Config:
def __init__(self):
self.device = "cuda:0"