Spaces:
Running
Running
File size: 3,249 Bytes
96134ee |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 74 75 76 77 78 79 80 81 82 83 84 85 86 87 88 89 90 91 |
import os
import json
import torch
version_config_paths = [os.path.join(version, size) for version in ["v1", "v2"] for size in ["32000.json", "40000.json", "48000.json"]]
def singleton(cls):
instances = {}
def get_instance(*args, **kwargs):
if cls not in instances: instances[cls] = cls(*args, **kwargs)
return instances[cls]
return get_instance
@singleton
class Config:
def __init__(self):
self.device = "cuda:0" if torch.cuda.is_available() else "cpu"
self.configs = json.load(open(os.path.join("main", "configs", "config.json"), "r"))
self.translations = self.multi_language()
self.json_config = self.load_config_json()
self.gpu_mem = None
self.per_preprocess = 3.7
self.is_half = self.is_fp16()
self.x_pad, self.x_query, self.x_center, self.x_max = self.device_config()
def multi_language(self):
try:
lang = self.configs.get("language", "en-US")
if len([l for l in os.listdir(os.path.join("assets", "languages")) if l.endswith(".json")]) < 1: raise FileNotFoundError("Không tìm thấy bất cứ gói ngôn ngữ nào(No package languages found)")
if not lang: lang = "en-US"
if lang not in self.configs["support_language"]: raise ValueError("Language not supported....")
lang_path = os.path.join("assets", "languages", f"{lang}.json")
if not os.path.exists(lang_path): lang_path = os.path.join("assets", "languages", "en-US.json")
with open(lang_path, encoding="utf-8") as f:
translations = json.load(f)
except json.JSONDecodeError:
print(self.translations["empty_json"].format(file=lang))
pass
return translations
def is_fp16(self):
fp16 = self.configs.get("fp16", False)
if self.device in ["cpu", "mps"] and fp16:
self.configs["fp16"] = False
fp16 = False
with open(os.path.join("main", "configs", "config.json"), "w") as f:
json.dump(self.configs, f, indent=4)
if not fp16: self.preprocess_per = 3.0
return fp16
def load_config_json(self):
configs = {}
for config_file in version_config_paths:
try:
with open(os.path.join("main", "configs", config_file), "r") as f:
configs[config_file] = json.load(f)
except json.JSONDecodeError:
print(self.translations["empty_json"].format(file=config_file))
pass
return configs
def device_config(self):
if self.device.startswith("cuda"): self.set_cuda_config()
elif self.has_mps(): self.device = "mps"
else: self.device = "cpu"
if self.gpu_mem is not None and self.gpu_mem <= 4:
self.preprocess_per = 3.0
return 1, 5, 30, 32
return (3, 10, 60, 65) if self.is_half else (1, 6, 38, 41)
def set_cuda_config(self):
i_device = int(self.device.split(":")[-1])
self.gpu_mem = torch.cuda.get_device_properties(i_device).total_memory // (1024**3)
def has_mps(self):
return torch.backends.mps.is_available()
|