updated info and default values

.idea/inspectionProfiles/Project_Default.xml

@@ -0,0 +1,15 @@
+<component name="InspectionProjectProfileManager">
+  <profile version="1.0">
+    <option name="myName" value="Project Default" />
+    <inspection_tool class="PyPep8NamingInspection" enabled="true" level="WEAK WARNING" enabled_by_default="true">
+      <option name="ignoredErrors">
+        <list>
+          <option value="N801" />
+          <option value="N806" />
+          <option value="N802" />
+          <option value="N803" />
+        </list>
+      </option>
+    </inspection_tool>
+  </profile>
+</component>

.idea/inspectionProfiles/profiles_settings.xml

@@ -0,0 +1,6 @@
+<component name="InspectionProjectProfileManager">
+  <settings>
+    <option name="USE_PROJECT_PROFILE" value="false" />
+    <version value="1.0" />
+  </settings>
+</component>

.idea/misc.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="PyCharmProfessionalAdvertiser">
+    <option name="shown" value="true" />
+  </component>
+</project>

.idea/vcs.xml

@@ -0,0 +1,6 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="VcsDirectoryMappings">
+    <mapping directory="$PROJECT_DIR$" vcs="Git" />
+  </component>
+</project>

.idea/workspace.xml

@@ -0,0 +1,66 @@
+<?xml version="1.0" encoding="UTF-8"?>
+<project version="4">
+  <component name="ChangeListManager">
+    <list default="true" id="0a8b71eb-dee1-421b-9887-746d3011c8ae" name="更改" comment="">
+      <change beforePath="$PROJECT_DIR$/app.py" beforeDir="false" afterPath="$PROJECT_DIR$/app.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/text/japanese.py" beforeDir="false" afterPath="$PROJECT_DIR$/text/japanese.py" afterDir="false" />
+      <change beforePath="$PROJECT_DIR$/text/japanese_bert.py" beforeDir="false" afterPath="$PROJECT_DIR$/text/japanese_bert.py" afterDir="false" />
+    </list>
+    <option name="SHOW_DIALOG" value="false" />
+    <option name="HIGHLIGHT_CONFLICTS" value="true" />
+    <option name="HIGHLIGHT_NON_ACTIVE_CHANGELIST" value="false" />
+    <option name="LAST_RESOLUTION" value="IGNORE" />
+  </component>
+  <component name="Git.Settings">
+    <option name="RECENT_GIT_ROOT_PATH" value="$PROJECT_DIR$" />
+  </component>
+  <component name="ProjectId" id="2WwCDjCv4hxVRoQDejQZgjz3Z96" />
+  <component name="ProjectViewState">
+    <option name="hideEmptyMiddlePackages" value="true" />
+    <option name="showLibraryContents" value="true" />
+  </component>
+  <component name="PropertiesComponent">
+    <property name="RunOnceActivity.OpenProjectViewOnStart" value="true" />
+    <property name="RunOnceActivity.ShowReadmeOnStart" value="true" />
+    <property name="last_opened_file_path" value="$PROJECT_DIR$" />
+  </component>
+  <component name="RunManager">
+    <configuration name="app" type="PythonConfigurationType" factoryName="Python" temporary="true" nameIsGenerated="true">
+      <module name="Umamusume-DeBERTa-VITS2-TTS-JP" />
+      <option name="INTERPRETER_OPTIONS" value="" />
+      <option name="PARENT_ENVS" value="true" />
+      <envs>
+        <env name="PYTHONUNBUFFERED" value="1" />
+      </envs>
+      <option name="SDK_HOME" value="C:\Users\25749\AppData\Local\Programs\Python\Python39\python.exe" />
+      <option name="WORKING_DIRECTORY" value="$PROJECT_DIR$" />
+      <option name="IS_MODULE_SDK" value="false" />
+      <option name="ADD_CONTENT_ROOTS" value="true" />
+      <option name="ADD_SOURCE_ROOTS" value="true" />
+      <option name="SCRIPT_NAME" value="$PROJECT_DIR$/app.py" />
+      <option name="PARAMETERS" value="" />
+      <option name="SHOW_COMMAND_LINE" value="false" />
+      <option name="EMULATE_TERMINAL" value="false" />
+      <option name="MODULE_MODE" value="false" />
+      <option name="REDIRECT_INPUT" value="false" />
+      <option name="INPUT_FILE" value="" />
+      <method v="2" />
+    </configuration>
+    <recent_temporary>
+      <list>
+        <item itemvalue="Python.app" />
+      </list>
+    </recent_temporary>
+  </component>
+  <component name="SpellCheckerSettings" RuntimeDictionaries="0" Folders="0" CustomDictionaries="0" DefaultDictionary="应用程序级" UseSingleDictionary="true" transferred="true" />
+  <component name="TaskManager">
+    <task active="true" id="Default" summary="默认任务">
+      <changelist id="0a8b71eb-dee1-421b-9887-746d3011c8ae" name="更改" comment="" />
+      <created>1697634137773</created>
+      <option name="number" value="Default" />
+      <option name="presentableId" value="Default" />
+      <updated>1697634137773</updated>
+    </task>
+    <servers />
+  </component>
+</project>

app.py

@@ -192,7 +192,7 @@ if __name__ == "__main__":
                     choices=speakers, value=speakers[0], label="Speaker"
                 )
                 sdp_ratio = gr.Slider(
-                    minimum=0, maximum=1, value=0.3, step=0.05, label="SDP Ratio (个人认为更大的SDP Ratio会产生\"感情更强烈的\"语音) (I personally believe that a larger SDP Ratio will make the generated voice \"emotionally stronger\")"
+                    minimum=0, maximum=1, value=0.2, step=0.05, label="SDP Ratio (个人认为更大的SDP Ratio会产生\"感情更强烈的\"语音) (I personally believe that a larger SDP Ratio will make the generated voice \"emotionally stronger\")"
                 )
                 noise_scale = gr.Slider(
                     minimum=0.1, maximum=2, value=0.6, step=0.05, label="Noise Scale"

info.md

@@ -50,7 +50,7 @@
 
 ❤ Thanks to **SUSTech Center for Computational Science and Engineering**. ❤ This model is trained on A100 (40GB) x 2 with **batch size 32** in total.
 
-💪 This model has been trained for **1 cycle, 90K steps (=60 epoch),** currently. 💪
+💪 This model has been trained for **1 cycle, 180K steps (=120 epoch),** currently. 💪
 
 📕 This work uses linear with warmup **(7.5% of total steps)** LR scheduler with  ` max_lr=1e-4`. 📕
 

text/japanese.py

@@ -640,6 +640,7 @@ def distribute_phone(n_phone, n_word):
 
 
 tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese")
+# tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese-with-auto-jumanpp")
 
 
 def g2p(norm_text):
@@ -666,6 +667,7 @@ def g2p(norm_text):
 
 if __name__ == "__main__":
     tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese")
+    # tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese-with-auto-jumanpp")
     text = "張り切って行こう"
     from text.japanese_bert import get_bert_feature
 

text/japanese_bert.py

@@ -4,6 +4,7 @@ import sys
 from text.japanese import text2sep_kata
 
 tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese")
+# tokenizer = AutoTokenizer.from_pretrained("ku-nlp/deberta-v2-base-japanese-with-auto-jumanpp")
 
 models = dict()
 
@@ -28,6 +29,7 @@ def get_bert_feature_with_token(tokens, word2ph, device=None):
     if device not in models.keys():
         models[device] = AutoModelForMaskedLM.from_pretrained(
             "ku-nlp/deberta-v2-base-japanese"
+            # "ku-nlp/deberta-v2-base-japanese-with-auto-jumanpp"
         ).to(device)
     with torch.no_grad():
         inputs = torch.tensor(tokens).to(device).unsqueeze(0)