Skip to content

Instantly share code, notes, and snippets.

View litagin02's full-sized avatar

litagin02

View GitHub Profile
from pathlib import Path
from typing import Any, Optional
import tqdm
from torch.utils.data import Dataset
# HF pipelineで進捗表示をするために必要なDatasetクラス
class StrListDataset(Dataset[str]):
def __init__(self, original_list: list[str]) -> None:
@litagin02
litagin02 / pattern.py
Last active March 10, 2024 11:38
笑い声・感動詞を文字列から識別するやつ
import re
import jaconv
# お好きな日本語文章正規化をimport, jaconv.normalizeとかでいいかも
from text import text_normalize
import regex
import pyopenjtalk
# 「ー」と「っ」を取り除いた文章に対するひらがなの笑い声の正規表現
warai_pattern = (
r"("
@litagin02
litagin02 / safetensors.ipynb
Last active January 18, 2024 01:51
学習したpthファイルから事前学習モデルsafetensorsを作るやつ
Sorry, something went wrong. Reload?
Sorry, we cannot display this file.
Sorry, this file is invalid so it cannot be displayed.
@litagin02
litagin02 / simple_merge.py
Last active February 25, 2024 16:13
Bert-VITS2のモデルマージするやつ(声音・感情表現それぞれを取っ替えたり混ぜたり)
import os
import gradio as gr
import torch
from infer import get_net_g, infer
import utils
voice_keys = ["dec", "flow"]
speech_style_keys = ["enc_p"]