Skip to content

Commit cff3b4e

Browse files
chore(format): run black on dev (#791)
Co-authored-by: github-actions[bot] <github-actions[bot]@users.noreply.github.com>
1 parent d9e2eba commit cff3b4e

File tree

3 files changed

+4
-7
lines changed

3 files changed

+4
-7
lines changed

ChatTTS/model/dvae.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -251,7 +251,7 @@ def __call__(
251251
self, inp: torch.Tensor, mode: Literal["encode", "decode"] = "decode"
252252
) -> torch.Tensor:
253253
return super().__call__(inp, mode)
254-
254+
255255
@torch.inference_mode()
256256
def load_pretrained(self, filename: str, device: torch.device):
257257
state_dict_tensors = load_safetensors(filename)

ChatTTS/utils/io.py

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -17,6 +17,7 @@ def load_safetensors(filename: str):
1717
state_dict_tensors[k] = f.get_tensor(k)
1818
return state_dict_tensors
1919

20+
2021
def get_latest_modified_file(directory):
2122

2223
files = [os.path.join(directory, f) for f in os.listdir(directory)]

examples/web/webui.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -249,12 +249,8 @@ def make_audio(autoplay, stream):
249249
)
250250
parser.add_argument("--server_port", type=int, default=8080, help="server port")
251251
parser.add_argument("--root_path", type=str, help="root path")
252-
parser.add_argument(
253-
"--custom_path", type=str, help="custom model path"
254-
)
255-
parser.add_argument(
256-
"--coef", type=str, help="custom dvae coefficient"
257-
)
252+
parser.add_argument("--custom_path", type=str, help="custom model path")
253+
parser.add_argument("--coef", type=str, help="custom dvae coefficient")
258254
args = parser.parse_args()
259255

260256
logger.info("loading ChatTTS model...")

0 commit comments

Comments
 (0)